Hugging Face's logo

Spaces:
neovalle
/
SystemPromptTestsGPU
Running on Zero

App Files Community
neovalle commited on
Commit
774c640
·
verified ·
1 Parent(s): 3d59343

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
14
  DEFAULT_MODELS = [
15
  "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
16
  "Qwen/Qwen2.5-1.5B-Instruct",
17
- "neovalle/tinyllama-h4rmony-dpo",
18
  ]
19
 
20
  _MODEL_CACHE = {} # cache: model_id -> (tokenizer, model)
 
14
  DEFAULT_MODELS = [
15
  "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
16
  "Qwen/Qwen2.5-1.5B-Instruct",
17
+ "neovalle/tinyllama-1.1B-h4rmony-trained",
18
  ]
19
 
20
  _MODEL_CACHE = {} # cache: model_id -> (tokenizer, model)