Fix bug

ModelTC · May 14, 2024 · 24c97b9 · 24c97b9
1 parent d92a51c
commit 24c97b9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/llmc/compression/quantization/omniq.py b/llmc/compression/quantization/omniq.py
@@ -85,10 +85,10 @@ def add_quant_config(self):
  else:
  self.load_clip = False
 
- if self.load_clip:
- assert "clip_path" in self.quant_config["special"]
- self.clip_path = self.quant_config["special"]["clip_path"]
- self.weight_clips = torch.load(self.clip_path)
+  if self.load_clip:
+  assert "clip_path" in self.quant_config["special"]
+  self.clip_path = self.quant_config["special"]["clip_path"]
+  self.weight_clips = torch.load(self.clip_path)
 
  if self.lwc:
  self.lwc_lr = self.quant_config["special"]["lwc_lr"]