wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-07-08 00:10:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

[NEW] Adding consistency in the URLs

Browse Source
This commit is contained in:
Nil-Andreu
2023-01-08 21:01:15 +01:00
parent 412736f52c
commit e241a8bf28
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/oasst_backend/api/v1/tasks.py
+1 -1
View File
@@ -285,7 +285,7 @@ async def tasks_interaction(
if not settings.DEBUG_SKIP_EMBEDDING_COMPUTATION:
try:
hugging_face_api = HuggingFaceAPI(
f"{HF_url.HUGGINGFACE_FEATURE_EXTRACTION.value}{HF_embeddingModel.MINILM.value}"
f"{HF_url.HUGGINGFACE_FEATURE_EXTRACTION.value}/{HF_embeddingModel.MINILM.value}"
)
embedding = await hugging_face_api.post(interaction.text)
pr.insert_message_embedding(
backend/oasst_backend/utils/hugging_face.py
+1 -1
View File
@@ -8,7 +8,7 @@ from oasst_shared.exceptions import OasstError, OasstErrorCode
class HF_url(str, Enum):
HUGGINGFACE_TOXIC_ROBERTA = ("https://api-inference.huggingface.co/models/unitary/multilingual-toxic-xlm-roberta",)
HUGGINGFACE_FEATURE_EXTRACTION = "https://api-inference.huggingface.co/pipeline/feature-extraction/"
HUGGINGFACE_FEATURE_EXTRACTION = "https://api-inference.huggingface.co/pipeline/feature-extraction"
class HF_embeddingModel(str, Enum):