Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
29 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
30 |
|
31 |
NUM_IMAGES_PER_PROMPT = 1
|
32 |
-
SAFETY_CHECKER =
|
33 |
if SAFETY_CHECKER:
|
34 |
from safety_checker import StableDiffusionSafetyChecker
|
35 |
from transformers import CLIPFeatureExtractor
|
|
|
29 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
30 |
|
31 |
NUM_IMAGES_PER_PROMPT = 1
|
32 |
+
SAFETY_CHECKER = True
|
33 |
if SAFETY_CHECKER:
|
34 |
from safety_checker import StableDiffusionSafetyChecker
|
35 |
from transformers import CLIPFeatureExtractor
|