Mikhil-jivus commited on
Commit
7e7821b
1 Parent(s): a9049fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ 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.bfloat16,
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
  token=access_token
42
  )
43
  model.eval()