Spaces:
kadirnar
/
Running on Zero

kadirnar commited on
Commit
8e4e52b
1 Parent(s): 991c987

Update demo_gradio.py

Browse files
Files changed (1) hide show
  1. demo_gradio.py +2 -2
demo_gradio.py CHANGED
@@ -38,7 +38,7 @@ from DPT.dpt.transforms import Resize, NormalizeImage, PrepareForNet
38
  """
39
  Get ZeST Ready
40
  """
41
- base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
42
  image_encoder_path = "models/image_encoder"
43
  ip_ckpt = "sdxl_models/ip-adapter_sdxl_vit-h.bin"
44
  controlnet_path = "diffusers/controlnet-depth-sdxl-1.0"
@@ -161,7 +161,7 @@ def greet(input_image, material_exemplar):
161
 
162
 
163
  num_samples = 1
164
- images = ip_model.generate(pil_image=ip_image, image=init_img, control_image=depth_map, mask_image=mask, controlnet_conditioning_scale=0.9, num_samples=num_samples, num_inference_steps=30, seed=42)
165
 
166
  return images[0]
167
 
 
38
  """
39
  Get ZeST Ready
40
  """
41
+ base_model_path = "Lykon/dreamshaper-xl-lightning"
42
  image_encoder_path = "models/image_encoder"
43
  ip_ckpt = "sdxl_models/ip-adapter_sdxl_vit-h.bin"
44
  controlnet_path = "diffusers/controlnet-depth-sdxl-1.0"
 
161
 
162
 
163
  num_samples = 1
164
+ images = ip_model.generate(guidance_scale=2, pil_image=ip_image, image=init_img, control_image=depth_map, mask_image=mask, controlnet_conditioning_scale=0.9, num_samples=num_samples, num_inference_steps=6, seed=42)
165
 
166
  return images[0]
167