We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2d06126 commit 4e3837eCopy full SHA for 4e3837e
intel_extension_for_pytorch/llm/utils.py
@@ -449,7 +449,7 @@ def _get_class_from_dynamic_module(
449
450
if code_revision is None and pretrained_model_name_or_path == repo_id:
451
code_revision = revision
452
- if code_revision is not None:
+ if code_revision is not None and code_revision != "main":
453
code_revision = "_" + code_revision[1:]
454
# And lastly we get the class inside our newly created module
455
final_module = _get_cached_module_file(
0 commit comments