Skip to content

Commit

Permalink
BUG: disable vicuna pytorch model (#225)
Browse files Browse the repository at this point in the history
  • Loading branch information
pangyoki authored Jul 20, 2023
1 parent aececfb commit 4e346e6
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions xinference/model/llm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -413,6 +413,8 @@ def install():
),
)

"""
# ggmlv3 model has the same model_name as pytorch model, comment it out in temporary
pytorch_vicuna_v1_3_name_generator = lambda model_size, quantization: (
f"lmsys/vicuna-{model_size}b-v1.3"
)
Expand All @@ -427,3 +429,4 @@ def install():
cls=VicunaCensoredPytorch,
),
)
"""

0 comments on commit 4e346e6

Please sign in to comment.