deps: lazy import optional dependencies gguf and torchvision (#4826)

This commit is contained in:
Juwan Yoo
2025-03-27 14:35:36 -07:00
committed by GitHub
parent b39532587b
commit 188105a21b
3 changed files with 25 additions and 3 deletions

View File

@@ -14,7 +14,6 @@ from abc import ABC, abstractmethod
from contextlib import contextmanager
from typing import Any, Dict, Generator, Iterable, List, Optional, Tuple, cast
import gguf
import huggingface_hub
import numpy as np
import torch
@@ -1155,6 +1154,17 @@ class GGUFModelLoader(BaseModelLoader):
See "Standardized tensor names" in
https://github.com/ggerganov/ggml/blob/master/docs/gguf.md for details.
"""
# only load the gguf module when needed
try:
import gguf
# FIXME: add version check for gguf
except ImportError as err:
raise ImportError(
"Please install gguf via `pip install gguf` to use gguf quantizer."
) from err
config = model_config.hf_config
model_type = config.model_type
# hack: ggufs have a different name than transformers