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/hidream/hidream_e1_model.py
+++ b/extensions_built_in/diffusion_models/hidream/hidream_e1_model.py
@@ -7,7 +7,7 @@ from toolkit.accelerator import unwrap_model
 import torch
 from toolkit.prompt_utils import PromptEmbeds
 from toolkit.config_modules import GenerateImageConfig
-from diffusers.models import HiDreamImageTransformer2DModel
+from toolkit.models.v2.diffusion_models.hidream import HiDreamImageTransformer2DModel

 import torch.nn.functional as F
 from PIL import Image