MCPcopy Create free account

hub / github.com/HiDream-ai/HiDream-I1 / types & classes

Types & classes21 in github.com/HiDream-ai/HiDream-I1

⨍ Functions 108◇ Types & classes 21
↓ 5 callersClassFeedForwardSwiGLU
hi_diffusers/models/attention.py:77
↓ 2 callersClassHiDreamAttention
hi_diffusers/models/attention.py:8
↓ 2 callersClassHiDreamAttnProcessor_flashattn
Attention processor used typically in processing the SD3-like self-attention projections.
hi_diffusers/models/attention_processor.py:44
↓ 2 callersClassHiDreamImageBlock
hi_diffusers/models/transformers/transformer_hidream_image.py:191
↓ 2 callersClassMOEFeedForwardSwiGLU
hi_diffusers/models/moe.py:97
↓ 1 callersClassEmbedND
hi_diffusers/models/embeddings.py:23
↓ 1 callersClassFlashFlowMatchEulerDiscreteSchedulerOutput
Output class for the scheduler's `step` function output. Args: prev_sample (`torch.FloatTensor` of shape `(batch_size, num_channels,
hi_diffusers/schedulers/flash_flow_match.py:33
↓ 1 callersClassHiDreamImagePipelineOutput
Output class for HiDreamImage pipelines. Args: images (`List[PIL.Image.Image]` or `np.ndarray`) List of denoised PIL ima
hi_diffusers/pipelines/hidream_image/pipeline_output.py:11
↓ 1 callersClassMoEGate
hi_diffusers/models/moe.py:34
↓ 1 callersClassOutEmbed
hi_diffusers/models/embeddings.py:93
↓ 1 callersClassPatchEmbed
hi_diffusers/models/embeddings.py:37
↓ 1 callersClassPooledEmbed
hi_diffusers/models/embeddings.py:60
↓ 1 callersClassTextProjection
hi_diffusers/models/transformers/transformer_hidream_image.py:21
↓ 1 callersClassTimestepEmbed
hi_diffusers/models/embeddings.py:75
ClassBlockType
hi_diffusers/models/transformers/transformer_hidream_image.py:30
ClassFlashFlowMatchEulerDiscreteScheduler
Euler scheduler. This model inherits from [`SchedulerMixin`] and [`ConfigMixin`]. Check the superclass documentation for the generic met
hi_diffusers/schedulers/flash_flow_match.py:46
ClassFlowUniPCMultistepScheduler
`UniPCMultistepScheduler` is a training-free framework designed for the fast sampling of diffusion models. This model inherits from [`Schedu
hi_diffusers/schedulers/fm_solvers_unipc.py:20
ClassHiDreamImagePipeline
hi_diffusers/pipelines/hidream_image/pipeline_hidream_image.py:111
ClassHiDreamImageSingleTransformerBlock
hi_diffusers/models/transformers/transformer_hidream_image.py:35
ClassHiDreamImageTransformer2DModel
hi_diffusers/models/transformers/transformer_hidream_image.py:230
ClassHiDreamImageTransformerBlock
hi_diffusers/models/transformers/transformer_hidream_image.py:106