mirror of
https://github.com/vladmandic/automatic
synced 2026-09-05 20:40:44 +02:00
CalamitousFelicitousness
d55af33d36
K3 and K5 share only the marketing name: different diffusers pipeline classes (Kandinsky3Pipeline vs Kandinsky5T2I/I2IPipeline), different denoiser classes (Kandinsky3UNet vs Kandinsky5Transformer3DModel), no shared scaffolding. Separate packages so each spec lives next to the arch it serves. K2.1 and K2.2 stay outside both packages since they go through KandinskyCombinedPipeline.from_pretrained directly with no spec.
14 lines
392 B
Python
14 lines
392 B
Python
"""Kandinsky 3.0 pipeline package.
|
|
|
|
Exports :data:`KANDINSKY3_UNET_SPEC`. Kandinsky 3 uses
|
|
:class:`Kandinsky3UNet` in the ``unet`` subfolder of the repo, hence
|
|
``subfolder='unet'`` instead of the default ``'transformer'``.
|
|
"""
|
|
|
|
import diffusers
|
|
|
|
from pipelines.native_transformer import TransformerSpec
|
|
|
|
|
|
KANDINSKY3_UNET_SPEC = TransformerSpec(cls=diffusers.Kandinsky3UNet, subfolder='unet')
|