remove unused lines

PiperOrigin-RevId: 607354714
google · Feb 15, 2024 · 5605af8 · 5605af8
1 parent c61583c
commit 5605af8
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/praxis/layers/quantization/linears.py b/praxis/layers/quantization/linears.py
@@ -113,7 +113,6 @@ def _get_weight_hparams(
         scale_sharding = None
     else:
       weight_sharding = wp.wt
-      scale_sharding = None
       if wp.wt is not None and len(wp.wt) > 1:
         scale_sharding = [wp.wt[1]]
       else:
@@ -327,7 +326,6 @@ def quantize_weight(self) -> NestedJTensor:
         'quantize_weight is called during serving for quantized model, please'
         ' set quantized config for the model.'
     )
-    theta = self.theta
     scale_name = 'w' + base_layer.QUANTIZED_SCALE_NAME_POSTFIX
 
     w = self.theta.w