jadechoghari
commited on
Commit
•
de1c669
1
Parent(s):
ca410d5
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -16,7 +16,7 @@ class MARModel(DiffusionPipeline):
|
|
16 |
super().__init__()
|
17 |
|
18 |
@torch.no_grad()
|
19 |
-
def
|
20 |
"""
|
21 |
This method downloads the model and VAE components,
|
22 |
then executes the forward pass based on the user's input.
|
|
|
16 |
super().__init__()
|
17 |
|
18 |
@torch.no_grad()
|
19 |
+
def __call__(self, *args, **kwargs):
|
20 |
"""
|
21 |
This method downloads the model and VAE components,
|
22 |
then executes the forward pass based on the user's input.
|