Hugging Face's logo

Spaces:
tushar-r-pawar
/
internlm2_5-7b-chat_with_airllm
Sleeping

App Files Community
tushar-r-pawar commited on
Commit
a513cc4
·
verified ·
1 Parent(s): 31a4022

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ model = AutoModelForCausalLM.from_pretrained(
31
  #model = AutoModelForCausalLM.from_pretrained(model_name)
32
 
33
  # Initialize AirLLM
34
- air_llm = airllm.AutoModel(model, tokenizer)
35
 
36
  # Streamlit app configuration
37
  st.set_page_config(
 
31
  #model = AutoModelForCausalLM.from_pretrained(model_name)
32
 
33
  # Initialize AirLLM
34
+ air_llm = airllm.AutoModel(model)
35
 
36
  # Streamlit app configuration
37
  st.set_page_config(