Skip to content

Commit f5144c1

Browse files
fix: add type safety checks for T5Gemma model initialization
1 parent 93b3228 commit f5144c1

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6451,8 +6451,14 @@ def __init__(self, *args, **kwargs):
64516451

64526452
# Initialize basic attributes manually
64536453
self.dir_model = args[0] if args else kwargs.get('dir_model')
6454+
if self.dir_model is None:
6455+
raise ValueError("dir_model is required")
64546456
self.ftype = args[1] if len(args) > 1 else kwargs.get('ftype')
6457+
if self.ftype is None:
6458+
raise ValueError("ftype is required")
64556459
self.fname_out = args[2] if len(args) > 2 else kwargs.get('fname_out')
6460+
if self.fname_out is None:
6461+
raise ValueError("fname_out is required")
64566462
self.is_big_endian = kwargs.get('is_big_endian', False)
64576463
self.endianess = gguf.GGUFEndian.BIG if self.is_big_endian else gguf.GGUFEndian.LITTLE
64586464
self.use_temp_file = kwargs.get('use_temp_file', False)
@@ -6466,6 +6472,8 @@ def __init__(self, *args, **kwargs):
64666472
if self.remote_hf_model_id is not None:
64676473
self.is_safetensors = True
64686474
def get_remote_tensors() -> Iterator[tuple[str, Tensor]]:
6475+
if self.remote_hf_model_id is None:
6476+
raise ValueError("remote_hf_model_id is required for remote models")
64696477
logger.info(f"Using remote model with HuggingFace id: {self.remote_hf_model_id}")
64706478
remote_tensors = gguf.utility.SafetensorRemote.get_list_tensors_hf_model(self.remote_hf_model_id)
64716479
self.tensor_names = set(name for name in remote_tensors.keys())

0 commit comments

Comments
 (0)