ALVHB95 commited on
Commit
f67b750
·
1 Parent(s): 71a1a1a

new model

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -179,18 +179,12 @@ qa_prompt = ChatPromptTemplate.from_template(
179
  # =============================
180
  # 4) LLM — local, token-free
181
  # =============================
182
- LOCAL_MODEL_ID = os.environ.get("LOCAL_LLM", "mistralai/Mixtral-8x7B-v0.1")
183
- tok = AutoTokenizer.from_pretrained(LOCAL_MODEL_ID)
184
- mdl = AutoModelForSeq2SeqLM.from_pretrained(LOCAL_MODEL_ID)
185
-
186
- gen = pipeline(
187
- task="text2text-generation",
188
- model=mdl,
189
- tokenizer=tok,
190
- do_sample=False, # deterministic; helps JSON adherence
191
- )
192
- llm = HuggingFacePipeline(pipeline=gen)
193
-
194
 
195
  # ===========================================
196
  # 5) Chain (memory + robust JSON extraction)
 
179
  # =============================
180
  # 4) LLM — local, token-free
181
  # =============================
182
+ llm = HuggingFaceHub( repo_id="mistralai/Mixtral-8x7B-v0.1",
183
+ task="text-generation",
184
+ model_kwargs={ "max_new_tokens": 2000,
185
+ "top_k": 30,
186
+ "temperature": 0.1,
187
+ "repetition_penalty": 1.03 }, )
 
 
 
 
 
 
188
 
189
  # ===========================================
190
  # 5) Chain (memory + robust JSON extraction)