Mikhil-jivus commited on
Commit
906992b
1 Parent(s): dbb94cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -37,7 +37,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_id,token=access_token)
37
  model = AutoModelForCausalLM.from_pretrained(
38
  model_id,
39
  device_map=device,
40
- torch_dtype=torch.float8,
 
41
  token=access_token
42
  )
43
  model.eval()
 
37
  model = AutoModelForCausalLM.from_pretrained(
38
  model_id,
39
  device_map=device,
40
+ #torch_dtype=torch.float8,
41
+ load_in_8bit=True,
42
  token=access_token
43
  )
44
  model.eval()