mskrt commited on
Commit
89b9678
verified
1 Parent(s): 8489f1e

added pipeline registration

Browse files
Files changed (1) hide show
  1. pipeline.py +7 -1
pipeline.py CHANGED
@@ -2,7 +2,7 @@ import random
2
  from typing import Callable, Dict, List, Optional
3
 
4
  import torch
5
- from diffusers import DiffusionPipeline
6
  from diffusers.configuration_utils import ConfigMixin
7
 
8
 
@@ -373,3 +373,9 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
373
  # Postprocess to generate the final output
374
  images = self.postprocess(latents)
375
  return images
 
 
 
 
 
 
 
2
  from typing import Callable, Dict, List, Optional
3
 
4
  import torch
5
+ from diffusers import DiffusionPipeline, DiffusionPipelineRegistry
6
  from diffusers.configuration_utils import ConfigMixin
7
 
8
 
 
373
  # Postprocess to generate the final output
374
  images = self.postprocess(latents)
375
  return images
376
+
377
+
378
+ DiffusionPipelineRegistry.register_pipeline(
379
+ task="superdiff-and", # Custom task name
380
+ pipeline_class=SuperDiffPipeline
381
+ )