kouki321 commited on
Commit
03fbdfd
·
verified ·
1 Parent(s): ea6d2c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -92,7 +92,8 @@ def clean_up(cache, origin_len):
92
 
93
  # Initialize model and tokenizer
94
  def load_model_and_tokenizer():
95
- model_name = "tiiuae/falcon-rw-1b"
 
96
  #"deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
97
  #"facebook/opt-125m"
98
  # Load tokenizer and model from disk (without trust_remote_code)
 
92
 
93
  # Initialize model and tokenizer
94
  def load_model_and_tokenizer():
95
+ model_name = "Locutusque/TinyMistral-248M"
96
+ #"tiiuae/falcon-rw-1b"
97
  #"deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
98
  #"facebook/opt-125m"
99
  # Load tokenizer and model from disk (without trust_remote_code)