Update main.py

main.py

@@ -6,7 +6,7 @@ from peft import PeftModel, get_peft_config
 import json
 import torch
 
-'''
+
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 
@@ -33,7 +33,7 @@ model = base_model.to(device)
 
 # 加载 tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-'''
+
 
 def format_instruction(task, text):
     string = f"""### 指令: