mazenlhm commited on
Commit
9e3d6a4
1 Parent(s): 9cace16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model = AutoModelForCausalLM.from_pretrained(
11
  return_dict=True,
12
  load_in_8bit=False,
13
  device_map=torch.device("cuda") if torch.cuda.is_available() else "cpu",
14
- load_in_8bit_fp32_cpu_offload=True
15
  )
16
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
17
 
 
11
  return_dict=True,
12
  load_in_8bit=False,
13
  device_map=torch.device("cuda") if torch.cuda.is_available() else "cpu",
14
+ #load_in_8bit_fp32_cpu_offload=True
15
  )
16
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
17