Spaces:

silk-road
/

ChatHaruhi-Qwen118k-Extended

Runtime error

silk-road commited on Dec 18, 2023

Commit

fd1d127

•

1 Parent(s): c3375eb

Update ChatHaruhi/Qwen118k2GPT.py

Files changed (1) hide show

ChatHaruhi/Qwen118k2GPT.py CHANGED Viewed

@@ -17,8 +17,7 @@ def initialize_Qwen2LORA(model):
         model_qwen = AutoModelForCausalLM.from_pretrained(
             model,
             device_map="auto",
-            trust_remote_code=True,
-            accelerate=False
         )
         model_qwen = model_qwen.eval()
         # model_qwen = PeftModel.from_pretrained(

         model_qwen = AutoModelForCausalLM.from_pretrained(
             model,
             device_map="auto",
+            trust_remote_code=True
         )
         model_qwen = model_qwen.eval()
         # model_qwen = PeftModel.from_pretrained(