Tonic commited on
Commit
558dcbc
·
1 Parent(s): e7f15bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
35
  model = transformers.AutoModelForCausalLM.from_pretrained(model_name,
36
  device_map="auto",
37
  torch_dtype=torch.bfloat16,
38
- load_in_4bit=True
39
  )
40
  model.eval()
41
 
 
35
  model = transformers.AutoModelForCausalLM.from_pretrained(model_name,
36
  device_map="auto",
37
  torch_dtype=torch.bfloat16,
38
+ # load_in_4bit=True
39
  )
40
  model.eval()
41