fix chameleonvision UT failure (huggingface#39646)

* fix chameleonvision UT failure

Signed-off-by: [email protected] <Yao Matrix>

* fix style

Signed-off-by: Yao, Matrix <[email protected]>

---------

Signed-off-by: [email protected] <Yao Matrix>
Signed-off-by: Yao, Matrix <[email protected]>
Co-authored-by: root <Yao Matrix>

Author: yao-matrix

src/transformers/models/chameleon/modeling_chameleon.py

@@ -817,7 +817,11 @@ class ChameleonPreTrainedModel(PreTrainedModel):
 )
 class ChameleonVQVAE(ChameleonPreTrainedModel):
     config: ChameleonVQVAEConfig
-    _no_split_modules = ["ChameleonVQVAEVectorQuantizer"]
+    _no_split_modules = [
+        "ChameleonVQVAEVectorQuantizer",
+        "ChameleonVQVAEEncoderAttnBlock",
+        "ChameleonVQVAEEncoderResnetBlock",
+    ]
 
     def __init__(self, config: ChameleonVQVAEConfig):
         super().__init__(config)