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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(
15
  offload_folder="off_load",
16
  trust_remote_code=True,
17
  # torch_dtype="auto",
 
18
  )
19
  tokenizer = AutoTokenizer.from_pretrained(checkpoint,
20
  trust_remote_code=True,
 
15
  offload_folder="off_load",
16
  trust_remote_code=True,
17
  # torch_dtype="auto",
18
+ from_tf=True,
19
  )
20
  tokenizer = AutoTokenizer.from_pretrained(checkpoint,
21
  trust_remote_code=True,