Ahmed Faiyaz
commited on
Commit
•
247ad4d
1
Parent(s):
19f4240
Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -103,6 +103,8 @@ class MukhOboyobPipeline(DiffusionPipeline, TextualInversionLoaderMixin, LoraLoa
|
|
103 |
requires_safety_checker: bool = True,
|
104 |
):
|
105 |
super().__init__()
|
|
|
|
|
106 |
|
107 |
if hasattr(scheduler.config, "steps_offset") and scheduler.config.steps_offset != 1:
|
108 |
deprecation_message = (
|
@@ -170,8 +172,6 @@ class MukhOboyobPipeline(DiffusionPipeline, TextualInversionLoaderMixin, LoraLoa
|
|
170 |
|
171 |
|
172 |
|
173 |
-
text_encoder=AutoModel.from_pretrained("csebuetnlp/banglabert")
|
174 |
-
tokenizer=AutoModel.from_pretrained("csebuetnlp/banglabert")
|
175 |
|
176 |
self.register_modules(
|
177 |
vae=vae,
|
|
|
103 |
requires_safety_checker: bool = True,
|
104 |
):
|
105 |
super().__init__()
|
106 |
+
text_encoder=AutoModel.from_pretrained("csebuetnlp/banglabert")
|
107 |
+
tokenizer=AutoModel.from_pretrained("csebuetnlp/banglabert")
|
108 |
|
109 |
if hasattr(scheduler.config, "steps_offset") and scheduler.config.steps_offset != 1:
|
110 |
deprecation_message = (
|
|
|
172 |
|
173 |
|
174 |
|
|
|
|
|
175 |
|
176 |
self.register_modules(
|
177 |
vae=vae,
|