Hugging Face's logo

Spaces:
wdndev
/
tiny_llm_sft_92m_demo
Running

App Files Community
wdndev commited on
Commit
5e0f5b9
·
1 Parent(s): 321af35

req

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -0
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 = "./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 = "wdndev/tiny_llm_sft_92m"
11
 
12
  @st.cache_resource
13
  def load_model_tokenizer():
requirements.txt CHANGED
@@ -2,4 +2,5 @@ numpy
2
  torch
3
  torchvision
4
  torchaudio
 
5
  transformers
 
2
  torch
3
  torchvision
4
  torchaudio
5
+ accelerate
6
  transformers