Hugging Face's logo

Spaces:
tayyab-077
/
tayyab-chatbot
Sleeping

App Files Community
tayyab-077 commited on
Commit
6f535b2
·
1 Parent(s): e1aac58

updated

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, use_fast=True, token=HF_TO
27
  llm = AutoModelForCausalLM.from_pretrained(
28
  MODEL_PATH,
29
  device_map="cpu",
30
- torch_dtype="auto",
31
  token=HF_TOKEN
32
  )
33
 
 
27
  llm = AutoModelForCausalLM.from_pretrained(
28
  MODEL_PATH,
29
  device_map="cpu",
30
+ dtype="auto",
31
  token=HF_TOKEN
32
  )
33