daydayup1225 commited on
Commit
2613d29
·
1 Parent(s): 0b7aaec

Update app_modules/utils.py

Browse files
Files changed (1) hide show
  1. app_modules/utils.py +2 -2
app_modules/utils.py CHANGED
@@ -446,8 +446,8 @@ def load_finetune_tokenizer_and_model(base_model_name_or_path, load_8bit=False):
446
  base_model_name_or_path, device_map={"": device}, low_cpu_mem_usage=True
447
  )
448
 
449
- if not load_8bit:
450
- model.half() # seems to fix bugs for some users.
451
 
452
  model.eval()
453
  return tokenizer, model, device
 
446
  base_model_name_or_path, device_map={"": device}, low_cpu_mem_usage=True
447
  )
448
 
449
+ # if not load_8bit:
450
+ # model.half() # seems to fix bugs for some users.
451
 
452
  model.eval()
453
  return tokenizer, model, device