Update main.py

2026-07-22 11:30:30 +08:00 · 2026-07-22 11:30:30 +08:00 · 13953a8914
commit 13953a8914
parent a4aa514733
1 changed files with 1 additions and 2 deletions
--- a/Petals-to-the-Metal-ViT-Lora/main.py
+++ b/Petals-to-the-Metal-ViT-Lora/main.py
@ -88,8 +88,7 @@ for batch in train_dataloader:
    break
    plt.axis('off')
    plt.show()
-# %%
-!pip install torchao==0.16.0
+
 from peft import LoraConfig, get_peft_model

 # 加载预训练模型