Skip to content

Commit 292576e

Browse files
committed
Fix download retries
1 parent 1a87027 commit 292576e

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

surya/common/s3.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,4 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
129129
print(f"Failed to download {pretrained_model_name_or_path} after {retries} attempts.")
130130
raise e # Reraise exception after max retries
131131

132-
pretrained_model_name_or_path = local_path
133-
134-
return super().from_pretrained(pretrained_model_name_or_path, *args, **kwargs)
132+
return super().from_pretrained(local_path, *args, **kwargs)

0 commit comments

Comments
 (0)