Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion vllm/model_executor/model_loader/weight_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
from typing import Any, Callable, Optional, Union

import filelock
import gguf
import huggingface_hub.constants
import numpy as np
import torch
Expand All @@ -40,6 +39,11 @@
SafetensorsStreamer = runai_model_streamer.placeholder_attr(
"SafetensorsStreamer")

try:
import gguf
except ImportError:
gguf = PlaceholderModule("gguf")

try:
from fastsafetensors import SafeTensorsFileLoader, SingleGroup
except ImportError:
Expand Down