Spaces:

HTBris
/

zerogpu-sdxlTrainer

Running on Zero

zerogpu-sdxlTrainer / toolkit /models /v2 /diffusion_models /sd3.py

Deploy Ostris AI-Toolkit ZeroGPU Trainer (Illustrious XL / SDXL / FLUX.1)

855e1f5 3 days ago

331 Bytes

from diffusers import SD3Transformer2DModel as DiffusersSD3Transformer2DModel

from .._mixin import OstrisModelMixin


class SD3Transformer2DModel(DiffusersSD3Transformer2DModel, OstrisModelMixin):
    aitk_subfolder = "transformer"

    @classmethod
    def get_transformer_block_names(cls):
        return ["transformer_blocks"]

	from diffusers import SD3Transformer2DModel as DiffusersSD3Transformer2DModel

	from .._mixin import OstrisModelMixin


	class SD3Transformer2DModel(DiffusersSD3Transformer2DModel, OstrisModelMixin):
	aitk_subfolder = "transformer"

	@classmethod
	def get_transformer_block_names(cls):
	return ["transformer_blocks"]