|
@@ -1,4 +1,3 @@
|
|
|
-# make sure you're logged in with `huggingface-cli login`
|
|
|
|
|
import copy
|
|
import copy
|
|
|
import math
|
|
import math
|
|
|
import os.path
|
|
import os.path
|
|
@@ -14,10 +13,6 @@ from utils import get_token
|
|
|
class Unsafe(StableDiffusionPipeline):
|
|
class Unsafe(StableDiffusionPipeline):
|
|
|
_num_images = 4
|
|
_num_images = 4
|
|
|
|
|
|
|
|
- #def __init__(self, *args, **kwargs):
|
|
|
|
|
- # self._num_images = kwargs.pop('imgs', 4)
|
|
|
|
|
- # super().__init__(*args, **kwargs)
|
|
|
|
|
-
|
|
|
|
|
def _patch_checker(self):
|
|
def _patch_checker(self):
|
|
|
if getattr(self, '_patched', False):
|
|
if getattr(self, '_patched', False):
|
|
|
return
|
|
return
|