Skip to content

Commit 26f30e6

Browse files
committed
remove prints
1 parent 680a8ed commit 26f30e6

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1178,7 +1178,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
11781178
hf_quantizer.preprocess_model(
11791179
model=model, device_map=device_map, keep_in_fp32_modules=keep_in_fp32_modules
11801180
)
1181-
print(keep_in_fp32_modules)
1181+
11821182
# Now that the model is loaded, we can determine the device_map
11831183
device_map = _determine_device_map(
11841184
model, device_map, max_memory, torch_dtype, keep_in_fp32_modules, hf_quantizer

src/diffusers/pipelines/consisid/consisid_utils.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,11 @@
88
from torchvision.transforms import InterpolationMode
99
from torchvision.transforms.functional import normalize, resize
1010

11-
from ...utils import load_image
11+
from ...utils import get_logger, load_image
1212

1313

14+
logger = get_logger(__name__)
15+
1416
_insightface_available = importlib.util.find_spec("insightface") is not None
1517
_consisid_eva_clip_available = importlib.util.find_spec("consisid_eva_clip") is not None
1618
_facexlib_available = importlib.util.find_spec("facexlib") is not None
@@ -166,7 +168,7 @@ def process_face_embeddings(
166168

167169
# incase insightface didn't detect face
168170
if id_ante_embedding is None:
169-
print("fail to detect face using insightface, extract embedding on align face")
171+
logger.warning("Failed to detect face using insightface. Extracting embedding with align face")
170172
id_ante_embedding = face_helper_2.get_feat(align_face)
171173

172174
id_ante_embedding = torch.from_numpy(id_ante_embedding).to(device, weight_dtype) # torch.Size([512])

0 commit comments

Comments
 (0)