Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -125,7 +125,7 @@ et = time.time()
|
|
125 |
elapsed_time = et - st
|
126 |
print('Loading ControlNet took: ', elapsed_time, 'seconds')
|
127 |
st = time.time()
|
128 |
-
vae = AutoencoderKL.from_pretrained("
|
129 |
et = time.time()
|
130 |
elapsed_time = et - st
|
131 |
print('Loading VAE took: ', elapsed_time, 'seconds')
|
@@ -137,7 +137,7 @@ st = time.time()
|
|
137 |
# controlnet=[identitynet, zoedepthnet],
|
138 |
# torch_dtype=torch.float16)
|
139 |
|
140 |
-
pipe = StableDiffusionXLInstantIDImg2ImgPipeline.from_pretrained("SG161222/
|
141 |
vae=vae,
|
142 |
controlnet=[identitynet, zoedepthnet],
|
143 |
torch_dtype=torch.float16)
|
@@ -261,7 +261,6 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
|
|
261 |
face_emb = face_info['embedding']
|
262 |
face_kps = draw_kps(face_image, face_info['kps'])
|
263 |
except:
|
264 |
-
face_info = None
|
265 |
face_detected = False
|
266 |
face_emb = None
|
267 |
face_kps = face_image
|
@@ -461,7 +460,7 @@ def get_civitai_safetensors(link):
|
|
461 |
if(image["nsfwLevel"] == 1):
|
462 |
image_url = image["url"]
|
463 |
break
|
464 |
-
break
|
465 |
if(not model_link_download):
|
466 |
gr.Warning("We couldn't find a SDXL LoRA on the model you've sent")
|
467 |
raise Exception("We couldn't find a SDXL LoRA on the model you've sent")
|
|
|
125 |
elapsed_time = et - st
|
126 |
print('Loading ControlNet took: ', elapsed_time, 'seconds')
|
127 |
st = time.time()
|
128 |
+
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
129 |
et = time.time()
|
130 |
elapsed_time = et - st
|
131 |
print('Loading VAE took: ', elapsed_time, 'seconds')
|
|
|
137 |
# controlnet=[identitynet, zoedepthnet],
|
138 |
# torch_dtype=torch.float16)
|
139 |
|
140 |
+
pipe = StableDiffusionXLInstantIDImg2ImgPipeline.from_pretrained("SG161222/RealVisXL_V5.0",
|
141 |
vae=vae,
|
142 |
controlnet=[identitynet, zoedepthnet],
|
143 |
torch_dtype=torch.float16)
|
|
|
261 |
face_emb = face_info['embedding']
|
262 |
face_kps = draw_kps(face_image, face_info['kps'])
|
263 |
except:
|
|
|
264 |
face_detected = False
|
265 |
face_emb = None
|
266 |
face_kps = face_image
|
|
|
460 |
if(image["nsfwLevel"] == 1):
|
461 |
image_url = image["url"]
|
462 |
break
|
463 |
+
break
|
464 |
if(not model_link_download):
|
465 |
gr.Warning("We couldn't find a SDXL LoRA on the model you've sent")
|
466 |
raise Exception("We couldn't find a SDXL LoRA on the model you've sent")
|