File size: 877 Bytes
135b069
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from ..utils import deprecate
from .transformers.prior_transformer import PriorTransformer, PriorTransformerOutput


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)


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)