Update __init__.py

This commit is contained in:
Nicolas Patry 2023-08-08 12:09:18 +02:00 committed by GitHub
parent 1fdc88ee90
commit 71b8d349a5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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/"):