Spaces:

InvestmentResearchAI
/

LLM-ADE

Sleeping

WilliamGazeley commited on Jun 5

Commit

a494749

•

1 Parent(s): abe7082

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from concurrent.futures import ThreadPoolExecutor
 def load_models():
     base_pipe = pipeline(
         "text-generation",
-        model="TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T",
         max_length=512,
     )
     irai_pipe = pipeline(

 def load_models():
     base_pipe = pipeline(
         "text-generation",
+        model="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
         max_length=512,
     )
     irai_pipe = pipeline(