Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
prior_transformer.py 877 Bytes
Newer Older
Sayak Paul's avatar
[Core] move transformer scripts to `transformers` modules (#6747)  
Sayak Paul committed
1
2 
from ..utils import deprecate
from .transformers.prior_transformer import PriorTransformer, PriorTransformerOutput
Will Berman's avatar
kakaobrain unCLIP (#1428)  
Will Berman committed
3
4
Sayak Paul's avatar
[Core] move transformer scripts to `transformers` modules (#6747)  
Sayak Paul committed
5
6
7 
class PriorTransformerOutput(PriorTransformerOutput):
    deprecation_message = "Importing `PriorTransformerOutput` from `diffusers.models.prior_transformer` is deprecated and this will be removed in a future version. Please use `from diffusers.models.transformers.prior_transformer import PriorTransformerOutput`, instead."
    deprecate("PriorTransformerOutput", "0.29", deprecation_message)
Will Berman's avatar
kakaobrain unCLIP (#1428)  
Will Berman committed
8
9
Sayak Paul's avatar
[Core] move transformer scripts to `transformers` modules (#6747)  
Sayak Paul committed
10
11
12 
class PriorTransformer(PriorTransformer):
    deprecation_message = "Importing `PriorTransformer` from `diffusers.models.prior_transformer` is deprecated and this will be removed in a future version. Please use `from diffusers.models.transformers.prior_transformer import PriorTransformer`, instead."
    deprecate("PriorTransformer", "0.29", deprecation_message)