diff --git a/Petals-to-the-Metal-ViT-Lora/main.py b/Petals-to-the-Metal-ViT-Lora/main.py
index 448ed9e..d6d7bd5 100644
--- 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
 
 # 加载预训练模型