apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-04-28 10:26:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

change nli model (#167)

Browse source 
* change nli model

* Fix bug in hallucination

---------

Co-authored-by: Shuguang Chen <54548843+nehcgs@users.noreply.github.com>
This commit is contained in:
Co Tran 2024-10-09 19:10:08 -07:00 • committed by GitHub
parent 3b7c58698f
commit f9e3a052fc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
model_server/app/loader.py
View file

@ -30,7 +30,7 @@ def get_embedding_model(
def get_zero_shot_model(
model_name=os.getenv("ZERO_SHOT_MODELS", "katanemo/deberta-base-nli"),
model_name=os.getenv("ZERO_SHOT_MODELS", "katanemo/bart-large-mnli"),
):
print("Loading Zero-shot Model...")