Skip to content

Commit 74d14d6

Browse files
committed
Migrate transformers utils to utils package and expand functionality
1 parent c0d1267 commit 74d14d6

File tree

5 files changed

+1386
-492
lines changed

5 files changed

+1386
-492
lines changed

src/speculators/convert/eagle/utils.py

Lines changed: 0 additions & 180 deletions
This file was deleted.

src/speculators/utils/__init__.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,28 @@
11
from .auto_importer import AutoImporterMixin
22
from .pydantic_utils import PydanticClassRegistryMixin, ReloadableBaseModel
33
from .registry import RegistryMixin
4+
from .transformers_utils import (
5+
check_download_model_checkpoint,
6+
check_download_model_config,
7+
download_model_checkpoint_from_hub,
8+
load_model_checkpoint_config_dict,
9+
load_model_checkpoint_index_weight_files,
10+
load_model_checkpoint_state_dict,
11+
load_model_checkpoint_weight_files,
12+
load_model_config,
13+
)
414

515
__all__ = [
616
"AutoImporterMixin",
717
"PydanticClassRegistryMixin",
818
"RegistryMixin",
919
"ReloadableBaseModel",
20+
"check_download_model_checkpoint",
21+
"check_download_model_config",
22+
"download_model_checkpoint_from_hub",
23+
"load_model_checkpoint_config_dict",
24+
"load_model_checkpoint_index_weight_files",
25+
"load_model_checkpoint_state_dict",
26+
"load_model_checkpoint_weight_files",
27+
"load_model_config",
1028
]

0 commit comments

Comments
 (0)