Spaces:
Running
on
Zero
Running
on
Zero
GPU task aborted fix attempt
#12
by
cbensimon
HF staff
- opened
app.py
CHANGED
@@ -50,6 +50,8 @@ ip_model_plus = IPAdapterFaceIDPlus(pipe, image_encoder_path, ip_plus_ckpt, devi
|
|
50 |
app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
|
51 |
app.prepare(ctx_id=0, det_size=(640, 640))
|
52 |
|
|
|
|
|
53 |
@spaces.GPU(enable_queue=True)
|
54 |
def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
|
55 |
faceid_all_embeds = []
|
|
|
50 |
app = FaceAnalysis(name="buffalo_l", providers=['CPUExecutionProvider'])
|
51 |
app.prepare(ctx_id=0, det_size=(640, 640))
|
52 |
|
53 |
+
cv2.setNumThreads(1)
|
54 |
+
|
55 |
@spaces.GPU(enable_queue=True)
|
56 |
def generate_image(images, prompt, negative_prompt, preserve_face_structure, face_strength, likeness_strength, nfaa_negative_prompt, progress=gr.Progress(track_tqdm=True)):
|
57 |
faceid_all_embeds = []
|