From 13953a8914312d44aae5d2e2c60c3dafc5e48d55 Mon Sep 17 00:00:00 2001
From: yukun-hh <zhangyukunhh@gmail.com>
Date: Wed, 22 Jul 2026 11:30:30 +0800
Subject: [PATCH] Update main.py

---
 Petals-to-the-Metal-ViT-Lora/main.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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
 
 # 加载预训练模型