Hugging Face's logo

Spaces:
rynmurdock
/
generative_recsys
Running on A10G

App Files Community
5
rynmurdock commited on
Commit
97656de
•
1 Parent(s): e1ae8f0

Update patch_sdxl.py

Browse files
Files changed (1) hide show
  1. 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
- ).to(device)
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
+ )
49
 
50
  def check_nsfw_images(
51
  images: list[Image.Image],