jcleee commited on
Commit
5eb7bac
·
verified ·
1 Parent(s): 6bd8277

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -71,10 +71,10 @@ model_config = agent_config["model"]["data"]
71
 
72
  # === BUILD MODEL ===
73
  model = HfApiModel(
74
- model_id="mistralai/Mistral-7B-Instruct-v0.2",
75
  token=os.getenv("HF_TOKEN"),
76
  temperature=0.5,
77
- max_tokens=2048,
78
  custom_role_conversions=None,
79
  )
80
 
 
71
 
72
  # === BUILD MODEL ===
73
  model = HfApiModel(
74
+ model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
75
  token=os.getenv("HF_TOKEN"),
76
  temperature=0.5,
77
+ max_tokens=1024,
78
  custom_role_conversions=None,
79
  )
80