Added legacy paths
This commit is contained in:
11
toolkit/models/v2/diffusion_models/sd3.py
Normal file
11
toolkit/models/v2/diffusion_models/sd3.py
Normal file
@@ -0,0 +1,11 @@
|
||||
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"]
|
||||
Reference in New Issue
Block a user