checking

sayakpaul · sayakpaul · commit c9b680da2c84 · 2025-07-10T12:26:09.000+05:30
diff --git a/src/diffusers/models/model_loading_utils.py b/src/diffusers/models/model_loading_utils.py
@@ -388,8 +388,7 @@ def load_shard_files_with_threadpool(args_list):
 
     # Do not spawn anymore workers than you need
     num_workers = min(len(args_list), num_workers)
-    print(f"{num_workers=}")
-
+    
     logger.info(f"Loading model weights in parallel with {num_workers} workers...")
 
     error_msgs = []
diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py
@@ -1502,7 +1502,6 @@ def _load_pretrained_model(
             for shard_file in resolved_model_file
         ]
 
-        print(f"{is_parallel_loading_enabled=}, {is_file=}")
         if is_parallel_loading_enabled and is_file:
             offload_index, state_dict_index, _mismatched_keys, _error_msgs = load_shard_files_with_threadpool(
                 args_list

Original file line number	Diff line number	Diff line change
`@@ -1502,7 +1502,6 @@ def _load_pretrained_model(`
`1502`	`1502`	`for shard_file in resolved_model_file`
`1503`	`1503`	`]`
`1504`	`1504`
`1505`		`- print(f"{is_parallel_loading_enabled=}, {is_file=}")`
`1506`	`1505`	`if is_parallel_loading_enabled and is_file:`
`1507`	`1506`	`offload_index, state_dict_index, _mismatched_keys, _error_msgs = load_shard_files_with_threadpool(`
`1508`	`1507`	`args_list`