eagleswim commited on
Commit
648cf8e
·
verified ·
1 Parent(s): d9b37a9

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 = "Qwen/Qwen2.5-7B-Instruct"
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 = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
9
 
10
  # Initialize vLLM with CPU configuration