eagleswim commited on
Commit
ddd1d6b
·
verified ·
1 Parent(s): 3e783c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from vllm import LLM, SamplingParams
4
  from PIL import Image
5
 
6
  # Load the language model and tokenizer from Hugging Face
7
- model_name = "facebook/opt-125m"
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
 
10
  # Initialize vLLM with CPU configuration
 
4
  from PIL import Image
5
 
6
  # Load the language model and tokenizer from Hugging Face
7
+ model_name = "Qwen/Qwen2.5-7B-Instruct"
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
 
10
  # Initialize vLLM with CPU configuration