Skip to content

Commit 189c48e

Browse files
committed
formatting
Summary Signed-off-by: HDCharles <[email protected]>
1 parent 861614b commit 189c48e

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/llmcompressor/pytorch/model_load/helpers.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,7 @@ def load_safetensors_state_dict(file_path: str) -> Dict[str, torch.Tensor]:
144144
def copy_python_files_from_model_cache(model, save_path: str):
145145
config = model.config
146146
cache_path = None
147-
print("GGG", config._name_or_path)
148-
if hasattr(config, "_name_or_path") and len(config._name_or_path)>0:
147+
if hasattr(config, "_name_or_path") and len(config._name_or_path) > 0:
149148
import os
150149
import shutil
151150

src/llmcompressor/transformers/utils/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def infer_recipe_from_model_path(model_path: Union[str, Path]) -> Optional[str]:
9797
:return: The path to the recipe file if found, None otherwise.
9898
"""
9999
model_path = model_path.as_posix() if isinstance(model_path, Path) else model_path
100-
print("DD", model_path, os.path.isdir(model_path) , os.path.isfile(model_path))
100+
101101
if os.path.isdir(model_path) or os.path.isfile(model_path):
102102
# Model path is a local path to the model directory or file
103103
model_path = (

0 commit comments

Comments
 (0)