Skip to content

Commit 4e3837e

Browse files
authored
Fix download issue in phi4 (#3624)
1 parent 2d06126 commit 4e3837e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intel_extension_for_pytorch/llm/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -449,7 +449,7 @@ def _get_class_from_dynamic_module(
449449

450450
if code_revision is None and pretrained_model_name_or_path == repo_id:
451451
code_revision = revision
452-
if code_revision is not None:
452+
if code_revision is not None and code_revision != "main":
453453
code_revision = "_" + code_revision[1:]
454454
# And lastly we get the class inside our newly created module
455455
final_module = _get_cached_module_file(

0 commit comments

Comments
 (0)