Hugging Face's logo

Spaces:
yxbob
/
h94-IP-Adapter-FaceID
Running

App Files Community
1
yxbob commited on
Commit
ce0897a
•
1 Parent(s): 2aff7fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,7 +20,8 @@ safety_model_id = "CompVis/stable-diffusion-safety-checker"
20
  safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id)
21
  safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id)
22
 
23
- device = "cuda"
 
24
 
25
  noise_scheduler = DDIMScheduler(
26
  num_train_timesteps=1000,
 
20
  safety_feature_extractor = AutoFeatureExtractor.from_pretrained(safety_model_id)
21
  safety_checker = StableDiffusionSafetyChecker.from_pretrained(safety_model_id)
22
 
23
+ # cuda
24
+ device = "cpu"
25
 
26
  noise_scheduler = DDIMScheduler(
27
  num_train_timesteps=1000,