Spaces:
Sleeping
Sleeping
Commit
•
6037933
1
Parent(s):
23975d5
Update patch_sdxl.py
Browse files- patch_sdxl.py +1 -1
patch_sdxl.py
CHANGED
|
@@ -45,7 +45,7 @@ safety_checker = StableDiffusionSafetyChecker.from_pretrained(
|
|
| 45 |
).to(device)
|
| 46 |
feature_extractor = CLIPFeatureExtractor.from_pretrained(
|
| 47 |
"openai/clip-vit-base-patch32"
|
| 48 |
-
)
|
| 49 |
|
| 50 |
def check_nsfw_images(
|
| 51 |
images: list[Image.Image],
|
|
|
|
| 45 |
).to(device)
|
| 46 |
feature_extractor = CLIPFeatureExtractor.from_pretrained(
|
| 47 |
"openai/clip-vit-base-patch32"
|
| 48 |
+
).to(device)
|
| 49 |
|
| 50 |
def check_nsfw_images(
|
| 51 |
images: list[Image.Image],
|