Skip to content

Commit 062de3a

Browse files
fix: resolve type annotation issues in T5Gemma model initialization
1 parent f892c18 commit 062de3a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6447,9 +6447,11 @@ class T5GemmaModel(TextModel):
64476447

64486448
def __init__(self, *args, **kwargs):
64496449
# Load hyperparameters first to modify them for super().__init__()
6450-
dir_model: Path = args[0] if args else kwargs.get('dir_model')
6450+
dir_model = args[0] if args else kwargs.get('dir_model')
64516451
if dir_model is None:
64526452
raise ValueError("dir_model is required")
6453+
# Type assertion after None check
6454+
dir_model = Path(dir_model)
64536455

64546456
hparams = kwargs.get("hparams") or ModelBase.load_hparams(dir_model)
64556457
encoder_config = hparams.get("encoder", {})

0 commit comments

Comments
 (0)