Spaces:

dushuai112233
/

LLM

Paused

dushuai112233 commited on Jan 3

Commit

c604783

verified ·

1 Parent(s): 1707a69

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingA
 from peft import LoraConfig, get_peft_model, TaskType
 from datasets import load_dataset
 import torch
 def main():
     accelerator = Accelerator()#初始化

 from peft import LoraConfig, get_peft_model, TaskType
 from datasets import load_dataset
 import torch
+from accelerate import Accelerator
 def main():
     accelerator = Accelerator()#初始化