rename env variable.

InternLM · Nov 29, 2024 · c836242 · c836242
1 parent b49d6b6
commit c836242
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/pytorch/backends/dlinfer/linear.py b/lmdeploy/pytorch/backends/dlinfer/linear.py
@@ -16,7 +16,7 @@ def update_weights(self,
                        weight: torch.Tensor,
                        bias: Optional[torch.Tensor] = None):
         """update weights."""
-        if os.getenv('TORCH_MACA_NN_LAYOUT', 'False').lower() == 'true':
+        if os.getenv('DLINER_LINEAR_USE_NN_LAYOUT', '0') == '1':
             weight = weight.data.t().contiguous()
         return weight, bias