Phase 1

2026-08-27 11:53:08 -06:00
parent e8d9cf6d35
commit 8db198ec0a
51 changed files with 985 additions and 1273 deletions
--- a/extensions_built_in/diffusion_models/omnigen2/init.py
+++ b/extensions_built_in/diffusion_models/omnigen2/init.py
@@ -96,8 +96,8 @@ class OmniGen2Model(BaseModel):

        self.print_and_status_update("Loading transformer")

-        transformer = OmniGen2Transformer2DModel.from_pretrained(
-            model_path, subfolder="transformer", torch_dtype=torch.bfloat16
+        transformer = OmniGen2Transformer2DModel.load_model(
+            model_path, dtype=torch.bfloat16
        )

        if not self.low_vram: