Hugging Face's logo

Spaces:
Korakoe
/
convert-sd-ckpt-cpu
Runtime error

App Files Community
Korakoe commited on
Commit
c72f18a
•
1 Parent(s): 97adabb

convert to cpu only

Browse files
Files changed (1) hide show
  1. convert.py +1 -1
convert.py CHANGED
@@ -860,7 +860,7 @@ def convert_full_checkpoint(
860
 
861
  tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-large-patch14")
862
  safety_checker = StableDiffusionSafetyChecker.from_pretrained(
863
- "CompVis/stable-diffusion-safety-checker", device_map="auto"
864
  )
865
  feature_extractor = AutoFeatureExtractor.from_pretrained(
866
  "CompVis/stable-diffusion-safety-checker"
 
860
 
861
  tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-large-patch14")
862
  safety_checker = StableDiffusionSafetyChecker.from_pretrained(
863
+ "CompVis/stable-diffusion-safety-checker", device_map="cpu"
864
  )
865
  feature_extractor = AutoFeatureExtractor.from_pretrained(
866
  "CompVis/stable-diffusion-safety-checker"