github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-16 06:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

update config file

Browse Source
This commit is contained in:
csunny
2023-05-05 20:06:24 +08:00
parent a7364c9fe3
commit a164d2f156
3 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/model/vicuna_llm.py
View File

@@ -25,7 +25,7 @@ class VicunaRequestLLM(LLM):
"stop": stop
}
response = requests.post(
url=urljoin(vicuna_model_server, self.vicuna_generate_path),
url=urljoin(VICUNA_MODEL_SERVER, self.vicuna_generate_path),
data=json.dumps(params),
)
response.raise_for_status()
@@ -55,7 +55,7 @@ class VicunaEmbeddingLLM(BaseModel, Embeddings):
print("Sending prompt ", p)
response = requests.post(
url=urljoin(vicuna_model_server, self.vicuna_embedding_path),
url=urljoin(VICUNA_MODEL_SERVER, self.vicuna_embedding_path),
json={
"prompt": p
}