From 77db759d88ee23e498d8afe51326986020d302a0 Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Tue, 24 Jun 2025 13:15:31 +0300
Subject: [PATCH] Cleanup

---
 modules/shared_items.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/shared_items.py b/modules/shared_items.py
index cdc1f1980..baf247a35 100644
--- a/modules/shared_items.py
+++ b/modules/shared_items.py
@@ -41,7 +41,7 @@ pipelines = {
     'UniDiffuser': getattr(diffusers, 'UniDiffuserPipeline', None),
     'Amused': getattr(diffusers, 'AmusedPipeline', None),
     'HiDream': getattr(diffusers, 'HiDreamImagePipeline', None),
-    'OmniGenPipeline': getattr(diffusers, 'DiffusionPipeline', None),
+    'OmniGenPipeline': getattr(diffusers, 'OmniGenPipeline', None),
 
     # dynamically imported and redefined later
     'Meissonic': getattr(diffusers, 'DiffusionPipeline', None), # dynamically redefined and loaded in sd_models.load_diffuser