Merge pull request #69 from nateraw/bugfixes

Bugfixes
nateraw · Oct 10, 2022 · 959a501 · 959a501
2 parents fbf2711 + 8e9c2b9
commit 959a501
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 12 deletions.
diff --git a/stable_diffusion_videos/__init__.py b/stable_diffusion_videos/__init__.py
@@ -108,7 +108,7 @@ def __dir__():
             "make_video_pyav",
             "get_timesteps_arr",
         ],
-        "upsampling": ["PipelineRealESRGAN"],
+        "upsampling": ["RealESRGANModel"],
     },
 )
 

diff --git a/stable_diffusion_videos/stable_diffusion_pipeline.py b/stable_diffusion_videos/stable_diffusion_pipeline.py
@@ -808,14 +808,3 @@ def __init__(self):
 
     def forward(self, images=None, **kwargs):
         return images, [False]
-
-
-class NoCheck(ModelMixin):
-    """Can be used in place of safety checker. Use responsibly and at your own risk."""
-
-    def __init__(self):
-        super().__init__()
-        self.register_parameter(name="asdf", param=torch.nn.Parameter(torch.randn(3)))
-
-    def forward(self, images=None, **kwargs):
-        return images, [False]