We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 772906d commit d0bb3ebCopy full SHA for d0bb3eb
src/llmcompressor/entrypoints/oneshot.py
@@ -7,6 +7,8 @@
7
with various pipeline configurations for efficient model optimization.
8
"""
9
10
+from __future__ import annotations
11
+
12
import os
13
from datetime import datetime
14
from pathlib import Path
@@ -243,7 +245,7 @@ def oneshot(
243
245
clear_sparse_session: bool = False,
244
246
stage: str | None = None,
247
# Dataset arguments
- dataset: "str | Dataset | DatasetDict | None" = None,
248
+ dataset: str | Dataset | DatasetDict | None = None,
249
dataset_config_name: str | None = None,
250
dataset_path: str | None = None,
251
splits: str | list[str] | dict[str, str] | None = None,
0 commit comments