pipeline_utils.py

@classmethod
def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
    # Add debugging prints here
    print(f"Loading model from path: {pretrained_model_name_or_path}")
    print(f"Class: {cls}")
    print(f"Model args: {model_args}")
    print(f"Kwargs: {kwargs}")

    load_method_name = kwargs.pop("_from_pretrained_load_method", "from_config")
    if not isinstance(load_method_name, str):
        raise TypeError("load_method_name must be a string")

    load_method = getattr(cls, load_method_name)
    return load_method(pretrained_model_name_or_path, *model_args, **kwargs)