Update main.py · 13953a8914 - Kaggle - 烤吐司个人代码托管站

yukun-hh/Kaggle

Update main.py

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

									
										Petals-to-the-Metal-ViT-Lora/main.py
									
		+1
		-2
	
												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

					# 加载预训练模型

					# 加载预训练模型