mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-10 11:54:52 +00:00
Update __init__.py
This commit is contained in:
parent
1fdc88ee90
commit
71b8d349a5
@ -89,7 +89,7 @@ def get_model(
|
||||
revision,
|
||||
quantize=quantize,
|
||||
dtype=dtype,
|
||||
dtypetrust_remote_code=trust_remote_code,
|
||||
trust_remote_code=trust_remote_code,
|
||||
)
|
||||
|
||||
if model_id.startswith("bigcode/"):
|
||||
|
Loading…
Reference in New Issue
Block a user