primerz commited on
Commit
16d7182
·
verified ·
1 Parent(s): 054dd8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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("nubby/blessed-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,7 +137,7 @@ st = time.time()
137
  # controlnet=[identitynet, zoedepthnet],
138
  # torch_dtype=torch.float16)
139
 
140
- pipe = StableDiffusionXLInstantIDImg2ImgPipeline.from_pretrained("SG161222/RealVisXL_V4.0",
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")