Explorar o código

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

Xuan-Son Nguyen hai 8 meses
pai
achega
07c2e2f76c
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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")