Mistral的回答未完成

我正在开发一个Web应用程序,旨在根据用户上传到应用程序的文档提供的上下文来回答问题。问题在于,当我使用Mistral v0.2模型时,回答未能完成。它们在完成之前就被截断了。如果我使用OpenAI,回答就能正确完成。我使用以下提示:

template="""    ### [INST] 指令:根据上下文回答用户的问题。    如果在上下文中找不到合适的答案,请回答说没有足够的信息。    {context}    ### 问题:    {question}(用西班牙语回答) [/INST]    #"""    template="""    <s>[INST]    """prompt = PromptTemplate(        input_variables=['context','question'],        template = template    )vector = Chroma(client=db,        collection_name="coleccion4",        embedding_function=embeddings)retriever = vector.as_retriever(search_type="similarity", search_kwargs={"k":3})llm = HuggingFaceHub(        repo_id="mistralai/Mistral-7B-Instruct-v0.2",        model_kwargs = {"temperature":0.4},        huggingfacehub_api_token = apikey_huggingFace    )respuesta = rag_chain.invoke(user_question)

当我使用OpenAI运行代码时,我得到了这样的回应:enter image description here

但当我使用Mistral时,回答没有结束:enter image description here

为什么会这样?


回答:

我已经将max_new_tokens设置为2000,现在似乎可以正常工作了

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注