Update main.py

2026-07-22 11:30:30 +08:00
parent a4aa514733
commit 13953a8914
1 changed files with 1 additions and 2 deletions
@@ -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

 # 加载预训练模型