joermd commited on
Commit
c46bc0f
·
verified ·
1 Parent(s): 5b492c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import AutoModelForCausalLM
8
  from transformers import TextIteratorStreamer
9
  from threading import Thread
10
 
11
- model_name = 'AIDC-AI/Ovis1.6-Llama3.2-3B'
12
 
13
  # load model
14
  model = AutoModelForCausalLM.from_pretrained(model_name,
 
8
  from transformers import TextIteratorStreamer
9
  from threading import Thread
10
 
11
+ model_name = 'joermd/speedy-llama2'
12
 
13
  # load model
14
  model = AutoModelForCausalLM.from_pretrained(model_name,