Upload anytext.py
Browse files- anytext.py +2 -0
anytext.py
CHANGED
|
@@ -981,6 +981,8 @@ class AnyTextPipeline(
|
|
| 981 |
scheduler: KarrasDiffusionSchedulers,
|
| 982 |
safety_checker: StableDiffusionSafetyChecker,
|
| 983 |
feature_extractor: CLIPImageProcessor,
|
|
|
|
|
|
|
| 984 |
image_encoder: CLIPVisionModelWithProjection = None,
|
| 985 |
requires_safety_checker: bool = True,
|
| 986 |
):
|
|
|
|
| 981 |
scheduler: KarrasDiffusionSchedulers,
|
| 982 |
safety_checker: StableDiffusionSafetyChecker,
|
| 983 |
feature_extractor: CLIPImageProcessor,
|
| 984 |
+
text_embedding_module: TextEmbeddingModule = None,
|
| 985 |
+
auxiliary_latent_module: AuxiliaryLatentModule = None,
|
| 986 |
image_encoder: CLIPVisionModelWithProjection = None,
|
| 987 |
requires_safety_checker: bool = True,
|
| 988 |
):
|