Jelajahi Sumber

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

Xuan-Son Nguyen 8 bulan lalu
induk
melakukan
07c2e2f76c
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      convert_hf_to_gguf.py

+ 1 - 1
convert_hf_to_gguf.py

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