Update app.py

app.py

@@ -2,11 +2,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingA
 from peft import LoraConfig, get_peft_model, TaskType
 from datasets import load_dataset
 from torch.utils.tensorboard import SummaryWriter
-import os
 import torch
+import os
+
 def main():
     # 基础模型位置
-    model_name = "dushuai112233/Qwen2-1.5B-Instruct"
+    model_name = "dushuai112233/Qwen2-1.5B-Instruct"  # 使用你提供的模型
     # 设备
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -68,4 +69,4 @@ def main():
     model.save_pretrained('./output')
 
 if __name__ == '__main__':
-    main()
+    main()