wdndev commited on
Commit
574a1d4
·
1 Parent(s): 0207208
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers.generation.utils import GenerationConfig
7
  st.set_page_config(page_title="Tiny LLM 92M Demo")
8
  st.title("Tiny LLM 92M Demo")
9
 
10
- model_id = "wdndev/tiny_llm_sft_92m"
11
 
12
  @st.cache_resource
13
  def load_model_tokenizer():
 
7
  st.set_page_config(page_title="Tiny LLM 92M Demo")
8
  st.title("Tiny LLM 92M Demo")
9
 
10
+ model_id = "./tiny_llm_sft_92m"
11
 
12
  @st.cache_resource
13
  def load_model_tokenizer():