appinitdev commited on
Commit
42d6b3b
verified
1 Parent(s): 4b1c68a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ from ip_adapter.ip_adapter_faceid import IPAdapterFaceID
15
  from insightface.app import FaceAnalysis
16
 
17
  # Configuraci贸n
18
- device = "cuda"
19
  model_id = "runwayml/stable-diffusion-v1-5"
20
  ip_ckpt = "ip-adapter-faceid_sd15.bin"
21
 
@@ -23,7 +23,7 @@ ip_ckpt = "ip-adapter-faceid_sd15.bin"
23
  app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'])
24
  app.prepare(ctx_id=0, det_size=(640, 640))
25
 
26
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(device)
27
  ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
28
 
29
  def generate(image, prompt):
 
15
  from insightface.app import FaceAnalysis
16
 
17
  # Configuraci贸n
18
+ device = "cpu"
19
  model_id = "runwayml/stable-diffusion-v1-5"
20
  ip_ckpt = "ip-adapter-faceid_sd15.bin"
21
 
 
23
  app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'])
24
  app.prepare(ctx_id=0, det_size=(640, 640))
25
 
26
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32).to(device)
27
  ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
28
 
29
  def generate(image, prompt):