We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5c8206f commit 1c04035Copy full SHA for 1c04035
src/llmcompressor/observers/base.py
@@ -84,9 +84,7 @@ def get_global_scale(self, observed: torch.Tensor) -> torch.nn.Parameter:
84
"""
85
observed = observed.reshape((1, 1, -1)) # per tensor reshape
86
min_vals, max_vals = self.get_min_max(observed)
87
- global_scale = generate_gparam(min_vals, max_vals)
88
-
89
- return global_scale
+ return generate_gparam(min_vals, max_vals)
90
91
def _get_module_param(self, name: str) -> Optional[torch.nn.Parameter]:
92
if self.module is None:
0 commit comments