Hugging Face's logo

Spaces:
Coldog2333
/
llm-jp-3-demo
Build error

App Files Community
Coldog2333 commited on
Commit
8619a17
·
1 Parent(s): 53ab578

Test with TinyLlama and Qwen models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
5
 
6
  # 可选模型列表 (CPU 配置只保留小模型)
7
  AVAILABLE_MODELS = [
8
- "llm-jp/llm-jp-3.1-1.8b-instruct4",
9
- "llm-jp/llm-jp-3-1.8b-instruct",
10
  ]
11
 
12
  # 全局变量存储当前模型
 
5
 
6
  # 可选模型列表 (CPU 配置只保留小模型)
7
  AVAILABLE_MODELS = [
8
+ "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
9
+ "Qwen/Qwen2.5-0.5B-Instruct",
10
  ]
11
 
12
  # 全局变量存储当前模型