Fix diffusers 0 25 dev

src/infer_compiler_registry/register_diffusers/__init__.py

@@ -13,8 +13,10 @@
     Attention: AttentionOflow,
     AttnProcessor2_0: AttnProcessorOflow,
     LoRAAttnProcessor2_0: LoRAAttnProcessorOflow,
-    Transformer2DModel: Transformer2DModelOflow,
 }
+if Transformer2DModelOflow is not None:
+    torch2oflow_class_map.update({Transformer2DModel: Transformer2DModelOflow})
+
 register(torch2oflow_class_map=torch2oflow_class_map)
 
 

src/infer_compiler_registry/register_diffusers/transformer_2d_oflow.py

@@ -10,6 +10,7 @@
 transformed_diffusers = transform_mgr.transform_package("diffusers")
 
 diffusers_0220_v = version.parse("0.22.0")
+diffusers_02499_v = version.parse("0.24.99")
 diffusers_version = version.parse(importlib.metadata.version("diffusers"))
 
 if diffusers_version < diffusers_0220_v:
@@ -366,7 +367,7 @@ def forward(
                 return (output,)
 
             return Transformer2DModelOutput(sample=output)
-else:
+elif diffusers_version < diffusers_02499_v:
     ConfigMixin = transformed_diffusers.configuration_utils.ConfigMixin
     register_to_config = transformed_diffusers.configuration_utils.register_to_config
     ImagePositionalEmbeddings = (
@@ -810,3 +811,5 @@ def forward(
                 return (output,)
 
             return Transformer2DModelOutput(sample=output)
+else:
+    Transformer2DModel = None

src/onediff/infer_compiler/transform/builtin_transform.py

@@ -168,7 +168,8 @@ def init(self):
             logger.info(f"convert {type(obj)} to {type(of_obj)}")
         return of_obj
     except Exception as e:
-        logger.error(f"Unsupported type: {type(obj)} {e=}")
+        logger.warning(f"Unsupported type: {type(obj)} {e=}")
+        # raise NotImplementedError(f"Unsupported type: {obj}")
         return obj