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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -6
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, DiffusionPipelineRegistry
6
  from diffusers.configuration_utils import ConfigMixin
7
 
8
 
@@ -374,8 +374,3 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
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
- )
 
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
 
 
374
  images = self.postprocess(latents)
375
  return images
376