LPX55 commited on
Commit
520b1aa
·
1 Parent(s): ff57491

minor fixes

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -43,14 +43,9 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
43
  controlnet=model,
44
  variant="fp16",
45
  )
46
- pipe = StableDiffusionXLFillPipeline.from_pretrained(
47
- "GraydientPlatformAPI/lustify-lightning",
48
- torch_dtype=torch.float16,
49
- vae=vae,
50
- controlnet=model,
51
- )
52
  pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
53
  pipe.to("cuda")
 
54
 
55
  @spaces.GPU(duration=12)
56
  def fill_image(prompt, image, model_selection, paste_back):
@@ -198,6 +193,7 @@ def inpaint(prompt, image, inpaint_model, paste_back):
198
  vae=vae,
199
  controlnet=model,
200
  ).to("cuda")
 
201
  mask = Image.fromarray(image["mask"]).convert("L")
202
  image = Image.fromarray(image["image"])
203
  inpaint_final_prompt = f"score_9, score_8_up, score_7_up, {prompt}"
@@ -214,6 +210,7 @@ def outpaint(image, width, height, overlap_percentage, num_inference_steps, resi
214
  cnet_image = background.copy()
215
  cnet_image.paste(0, (0, 0), mask)
216
  final_prompt = f"score_9, score_8_up, score_7_up, {prompt_input} , high quality, 4k"
 
217
  (
218
  prompt_embeds,
219
  negative_prompt_embeds,
 
43
  controlnet=model,
44
  variant="fp16",
45
  )
 
 
 
 
 
 
46
  pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
47
  pipe.to("cuda")
48
+ print(pipe)
49
 
50
  @spaces.GPU(duration=12)
51
  def fill_image(prompt, image, model_selection, paste_back):
 
193
  vae=vae,
194
  controlnet=model,
195
  ).to("cuda")
196
+ print(f"Loaded new SDXL model: {pipe.config.model_name}")
197
  mask = Image.fromarray(image["mask"]).convert("L")
198
  image = Image.fromarray(image["image"])
199
  inpaint_final_prompt = f"score_9, score_8_up, score_7_up, {prompt}"
 
210
  cnet_image = background.copy()
211
  cnet_image.paste(0, (0, 0), mask)
212
  final_prompt = f"score_9, score_8_up, score_7_up, {prompt_input} , high quality, 4k"
213
+ print(f"Outpainting using SDXL model: {pipe.config.model_name}")
214
  (
215
  prompt_embeds,
216
  negative_prompt_embeds,