yukun-hh/Kaggle
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update main.py

Browse source
This commit is contained in:
yukun-hh 2026-07-22 11:30:30 +08:00
parent a4aa514733
commit 13953a8914
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
Petals-to-the-Metal-ViT-Lora/main.py
View file

@ -88,8 +88,7 @@ for batch in train_dataloader:
break break
plt.axis('off') plt.axis('off')
plt.show() plt.show()
# %%
!pip install torchao==0.16.0
from peft import LoraConfig, get_peft_model from peft import LoraConfig, get_peft_model
# 加载预训练模型 # 加载预训练模型