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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -15,12 +15,10 @@ model = AutoModelForCausalLM.from_pretrained(
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,
22
  torch_dtype="auto",
23
- from_tf=True,
24
  )
25
 
26
  # model = "tiiuae/FalconLite2"
 
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,
21
  torch_dtype="auto",
 
22
  )
23
 
24
  # model = "tiiuae/FalconLite2"