vsrinivas commited on
Commit
b30b88f
·
verified ·
1 Parent(s): a3e18cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ model = AutoModelForCausalLM.from_pretrained(
19
  )
20
  tokenizer = AutoTokenizer.from_pretrained(checkpoint,
21
  trust_remote_code=True,
22
- torch_dtype="auto"
 
23
  )
24
 
25
  # model = "tiiuae/FalconLite2"
 
19
  )
20
  tokenizer = AutoTokenizer.from_pretrained(checkpoint,
21
  trust_remote_code=True,
22
+ torch_dtype="auto",
23
+ from_tf=True,
24
  )
25
 
26
  # model = "tiiuae/FalconLite2"