Hugging Face's logo

Spaces:
Aditya757864
/
chatbot
Sleeping

App Files Community
Aditya757864 commited on
Commit
e8d038f
·
verified ·
1 Parent(s): 39f15b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ import os
14
  import tempfile
15
 
16
  model = AutoModelForSeq2SeqLM.from_pretrained("LaMini-Flan-T5-783M")
17
- tokenizer = AutoTokenizer.from_pretrained("LaMini-Flan-T5-783M", device_map = 'auto', torch_dtype = torch.float32)
18
 
19
 
20
  checkpoint = "LaMini-Flan-T5-783M"
 
14
  import tempfile
15
 
16
  model = AutoModelForSeq2SeqLM.from_pretrained("LaMini-Flan-T5-783M")
17
+ tokenizer = AutoTokenizer.from_pretrained("LaMini-Flan-T5-783M", device_map = 'auto')
18
 
19
 
20
  checkpoint = "LaMini-Flan-T5-783M"