Skip to content

Commit 5d6b840

Browse files
committed
delete debugging print in sinq integration
1 parent 559e1d9 commit 5d6b840

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

src/transformers/quantizers/quantizer_sinq.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -174,8 +174,6 @@ def validate_environment(self, dtype=None, device_map=None, weights_only=None, *
174174

175175
device_str = _normalize_cuda_device(getattr(cfg, "device", "auto"))
176176
_validate_cuda_device_str(device_str)
177-
178-
print(f'Device string is: {device_str}')
179177

180178
self._normalized_device_str = device_str
181179

@@ -244,7 +242,6 @@ def _process_model_before_weight_loading(self, model: nn.Module, **kwargs) -> tu
244242
compute_dtype = self.update_dtype(None)
245243
to_skip = set(cfg.modules_to_not_convert or [])
246244
device_str = getattr(self, "_normalized_device_str", _normalize_cuda_device(cfg.device))
247-
print(f'Device string in process model before_weights: {device_str}')
248245

249246
def _convert(m: nn.Module, prefix: str = ""):
250247
for child_name, child in list(m.named_children()):
@@ -283,8 +280,6 @@ def _process_model_after_weight_loading(
283280
torch.cuda.set_device(device)
284281
model = model
285282

286-
print(f'Device string in process model after weights: {device_str}')
287-
288283
placeholders: list[tuple[str, _SinqLoadTimeLinear, nn.Module]] = []
289284

290285
def _gather(m: nn.Module, prefix: str = "", parent: Optional[nn.Module] = None):

0 commit comments

Comments
 (0)