richardr1126 commited on
Commit
ec5de02
·
1 Parent(s): f7fa740

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ print(f"Starting to load the model {model_name}")
20
  m = AutoModelForCausalLM.from_pretrained(
21
  model_name,
22
  device_map=0,
23
- load_in_8bit=True,
24
  )
25
 
26
  m.config.pad_token_id = m.config.eos_token_id
 
20
  m = AutoModelForCausalLM.from_pretrained(
21
  model_name,
22
  device_map=0,
23
+ #load_in_8bit=True,
24
  )
25
 
26
  m.config.pad_token_id = m.config.eos_token_id