silk-road commited on
Commit
c3375eb
β€’
1 Parent(s): 1961350

Update ChatHaruhi/Qwen118k2GPT.py

Browse files
Files changed (1) hide show
  1. ChatHaruhi/Qwen118k2GPT.py +2 -1
ChatHaruhi/Qwen118k2GPT.py CHANGED
@@ -17,7 +17,8 @@ def initialize_Qwen2LORA(model):
17
  model_qwen = AutoModelForCausalLM.from_pretrained(
18
  model,
19
  device_map="auto",
20
- trust_remote_code=True
 
21
  )
22
  model_qwen = model_qwen.eval()
23
  # model_qwen = PeftModel.from_pretrained(
 
17
  model_qwen = AutoModelForCausalLM.from_pretrained(
18
  model,
19
  device_map="auto",
20
+ trust_remote_code=True,
21
+ accelerate=False
22
  )
23
  model_qwen = model_qwen.eval()
24
  # model_qwen = PeftModel.from_pretrained(