Update README.md
Browse files
README.md
CHANGED
@@ -499,6 +499,15 @@ Potential Fix: deepcompressor.quantizer.impl.scale.py
|
|
499 |
s = exp_s if lin_s.data is None else lin_s.extend(exp_s)
|
500 |
else:
|
501 |
s = lin_s
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
502 |
assert s.data is not None, "Scale tensor is None."
|
503 |
assert not s.data.isnan().any(), "Scale tensor contains NaN."
|
504 |
assert not s.data.isinf().any(), "Scale tensor contains Inf."
|
|
|
499 |
s = exp_s if lin_s.data is None else lin_s.extend(exp_s)
|
500 |
else:
|
501 |
s = lin_s
|
502 |
+
|
503 |
+
# Before the final assertions, add debugging and validation
|
504 |
+
if s.data is None:
|
505 |
+
# Log debugging information
|
506 |
+
print(f"Linear scale dtypes: {self.linear_scale_quant_dtypes}")
|
507 |
+
print(f"Exponent scale dtypes: {self.exponent_scale_quant_dtypes}")
|
508 |
+
if hasattr(lin_s, 'data') and lin_s.data is not None:
|
509 |
+
print(f"Linear scale data shape: {lin_s.data.shape}")
|
510 |
+
raise RuntimeError("Scale computation failed - resulting scale is None")
|
511 |
assert s.data is not None, "Scale tensor is None."
|
512 |
assert not s.data.isnan().any(), "Scale tensor contains NaN."
|
513 |
assert not s.data.isinf().any(), "Scale tensor contains Inf."
|