Quellcode durchsuchen

convert : correct typo image_mean --> image_std (#13208)

Xuan-Son Nguyen vor 8 Monaten
Ursprung
Commit
07c2e2f76c
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      convert_hf_to_gguf.py

+ 1 - 1
convert_hf_to_gguf.py

@@ -1103,7 +1103,7 @@ class VisionModel(ModelBase):
 
         # preprocessor config
         self.gguf_writer.add_vision_image_mean(self.preprocessor_config["image_mean"])
-        self.gguf_writer.add_vision_image_std(self.preprocessor_config["image_mean"])
+        self.gguf_writer.add_vision_image_std(self.preprocessor_config["image_std"])
 
     def write_vocab(self):
         raise ValueError("VisionModel does not support vocab writing")