[Model] Handle trust_remote_code for transformers backend (#32194)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
@@ -887,6 +887,7 @@ class _ModelRegistry:
|
||||
module,
|
||||
model_config.model,
|
||||
revision=model_config.revision,
|
||||
trust_remote_code=model_config.trust_remote_code,
|
||||
warn_on_fail=False,
|
||||
)
|
||||
|
||||
@@ -899,6 +900,7 @@ class _ModelRegistry:
|
||||
module,
|
||||
model_config.model,
|
||||
revision=model_config.revision,
|
||||
trust_remote_code=model_config.trust_remote_code,
|
||||
warn_on_fail=True,
|
||||
)
|
||||
if model_module is not None:
|
||||
|
||||
Reference in New Issue
Block a user