surkovvv commited on
Commit
359d23c
·
1 Parent(s): ccd7279

changed model to saiga_llama3_8b

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, StoppingCriteria,
4
  from threading import Thread
5
 
6
 
7
- tokenizer = AutoTokenizer.from_pretrained("togethercomputer/RedPajama-INCITE-Chat-3B-v1")
8
- model = AutoModelForCausalLM.from_pretrained("togethercomputer/RedPajama-INCITE-Chat-3B-v1") #, torch_dtype=torch.float16)
9
  model = model #.to('cuda')
10
 
11
 
 
4
  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