azrai99 commited on
Commit
8ba343a
1 Parent(s): 4ee9429

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def configure_quantization():
24
  @st.cache_resource
25
  def initialize_llm(hf_token):
26
  # quantization_config = configure_quantization()
27
- model_name = 'GeneZC/MiniChat-1.5-3B'
28
  return HuggingFaceLLM(
29
  model_name = model_name, #meta-llama/Meta-Llama-3-8B-Instruct meta-llama/Llama-2-7b-chat-hf #google/gemma-7b-it #HuggingFaceH4/zephyr-7b-beta #'GeneZC/MiniChat-2-3B''ericzzz/falcon-rw-1b-chat'
30
  tokenizer_name = model_name,
 
24
  @st.cache_resource
25
  def initialize_llm(hf_token):
26
  # quantization_config = configure_quantization()
27
+ model_name = 'TinyLlama/TinyLlama_v1.1'
28
  return HuggingFaceLLM(
29
  model_name = model_name, #meta-llama/Meta-Llama-3-8B-Instruct meta-llama/Llama-2-7b-chat-hf #google/gemma-7b-it #HuggingFaceH4/zephyr-7b-beta #'GeneZC/MiniChat-2-3B''ericzzz/falcon-rw-1b-chat'
30
  tokenizer_name = model_name,