jadechoghari commited on
Commit
de1c669
1 Parent(s): ca410d5

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -16,7 +16,7 @@ class MARModel(DiffusionPipeline):
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.
 
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.