Spaces:

yxbob
/

h94-IP-Adapter-FaceID

Runtime error

yxbob commited on Jan 31, 2024

Commit

ce0897a

verified ·

1 Parent(s): 2aff7fe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,8 @@ safety_model_id = "CompVis/stable-diffusion-safety-checker"
 safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id)
 safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id)
-device = "cuda"
 noise_scheduler = DDIMScheduler(
     num_train_timesteps=1000,

 safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id)
 safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id)
+# cuda
+device = "cpu"
 noise_scheduler = DDIMScheduler(
     num_train_timesteps=1000,