fix: make --help return before heavy runtime imports (#293)
This commit is contained in:
+15
-2
@@ -3,6 +3,20 @@
|
|||||||
|
|
||||||
# ruff: noqa: E402
|
# ruff: noqa: E402
|
||||||
|
|
||||||
|
import sys
|
||||||
|
|
||||||
|
from .config import Settings
|
||||||
|
|
||||||
|
|
||||||
|
def _is_help_invocation() -> bool:
|
||||||
|
args = sys.argv[1:]
|
||||||
|
return "-h" in args or "--help" in args
|
||||||
|
|
||||||
|
|
||||||
|
# Parse and handle CLI help before importing heavyweight ML/runtime dependencies.
|
||||||
|
if _is_help_invocation():
|
||||||
|
Settings() # ty:ignore[missing-argument]
|
||||||
|
|
||||||
from .progress import patch_tqdm
|
from .progress import patch_tqdm
|
||||||
|
|
||||||
# This patches tqdm class definitions, which must happen
|
# This patches tqdm class definitions, which must happen
|
||||||
@@ -13,7 +27,6 @@ import logging
|
|||||||
import math
|
import math
|
||||||
import os
|
import os
|
||||||
import random
|
import random
|
||||||
import sys
|
|
||||||
import time
|
import time
|
||||||
import warnings
|
import warnings
|
||||||
from dataclasses import asdict
|
from dataclasses import asdict
|
||||||
@@ -45,7 +58,7 @@ from rich.table import Table
|
|||||||
from rich.traceback import install
|
from rich.traceback import install
|
||||||
|
|
||||||
from .analyzer import Analyzer
|
from .analyzer import Analyzer
|
||||||
from .config import QuantizationMethod, Settings
|
from .config import QuantizationMethod
|
||||||
from .evaluator import Evaluator
|
from .evaluator import Evaluator
|
||||||
from .model import AbliterationParameters, Model, get_model_class
|
from .model import AbliterationParameters, Model, get_model_class
|
||||||
from .system import empty_cache, get_accelerator_info
|
from .system import empty_cache, get_accelerator_info
|
||||||
|
|||||||
Reference in New Issue
Block a user