add vae pt. 1

Author: a-r-r-o-w

scripts/convert_cosmos_to_diffusers.py

@@ -62,7 +62,9 @@ def rename_transformer_blocks_(key: str, state_dict: Dict[str, Any]):
     "pos_embedder.seq": remove_keys_,
 }
 
-VAE_KEYS_RENAME_DICT = {}
+VAE_KEYS_RENAME_DICT = {
+    "conv3d": "conv",
+}
 
 VAE_SPECIAL_KEYS_REMAP = {}