project-baize commited on
Commit
a56a74d
1 Parent(s): 8ed0878

Update app_modules/utils.py

Browse files
Files changed (1) hide show
  1. app_modules/utils.py +1 -2
app_modules/utils.py CHANGED
@@ -352,13 +352,12 @@ def load_tokenizer_and_model(base_model,adapter_model,load_8bit=False):
352
  )
353
  else:
354
  model = LlamaForCausalLM.from_pretrained(
355
- base_model, device_map={"": device}, low_cpu_mem_usage=True,torch_dtype=torch.float16
356
  )
357
  model = PeftModel.from_pretrained(
358
  model,
359
  adapter_model,
360
  device_map={"": device},
361
- torch_dtype=torch.float16
362
  )
363
 
364
  if not load_8bit:
 
352
  )
353
  else:
354
  model = LlamaForCausalLM.from_pretrained(
355
+ base_model, device_map={"": device}, low_cpu_mem_usage=True
356
  )
357
  model = PeftModel.from_pretrained(
358
  model,
359
  adapter_model,
360
  device_map={"": device},
 
361
  )
362
 
363
  if not load_8bit: