Skip to content

Commit f849c17

Browse files
committed
fix format
Signed-off-by: Zhiyu Cheng <[email protected]>
1 parent 446e135 commit f849c17

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

examples/llm_ptq/vlm_utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,5 +231,3 @@ def run_text_only_generation(model, tokenizer, question, generation_config, mode
231231
except Exception as e:
232232
print(f"Text-only generation failed: {e}")
233233
return None
234-
235-

modelopt/torch/export/model_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@
6060
{MODEL_NAME_TO_TYPE=}
6161
"""
6262

63-
__all__ = ["get_model_type", "is_multimodal_model", "get_language_model_from_vl"]
63+
__all__ = ["get_language_model_from_vl", "get_model_type", "is_multimodal_model"]
6464

6565

6666
def get_model_type(model):

modelopt/torch/export/unified_export_hf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@
4343
is_quantlinear,
4444
set_expert_quantizer_amax,
4545
)
46-
from .model_utils import get_language_model_from_vl, is_multimodal_model
4746
from .model_config import (
4847
KV_CACHE_FP8,
4948
KV_CACHE_NVFP4,
@@ -56,6 +55,7 @@
5655
QUANTIZATION_W4A8_AWQ,
5756
QUANTIZATION_W4A8_NVFP4_FP8,
5857
)
58+
from .model_utils import get_language_model_from_vl, is_multimodal_model
5959
from .plugins import export_spec_ckpt_config, export_spec_ckpt_state_dict, spec_opt_only
6060
from .quant_utils import (
6161
fuse_prequant_layernorm,

0 commit comments

Comments
 (0)