Hugging Face's logo

Spaces:
Zubik
/
chatbot-demo
Sleeping

App Files Community
surkovvv commited on
Commit
632b373
·
1 Parent(s): 359d23c

bloat half precision

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from threading import Thread
5
 
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("IlyaGusev/saiga_llama3_8b")
8
- model = AutoModelForCausalLM.from_pretrained("IlyaGusev/saiga_llama3_8b") #, torch_dtype=torch.float16)
9
  model = model #.to('cuda')
10
 
11
 
 
5
 
6
 
7
  tokenizer = AutoTokenizer.from_pretrained("IlyaGusev/saiga_llama3_8b")
8
+ model = AutoModelForCausalLM.from_pretrained("IlyaGusev/saiga_llama3_8b", torch_dtype=torch.bfloat16)
9
  model = model #.to('cuda')
10
 
11