kayfahaarukku commited on
Commit
f13e2d2
1 Parent(s): 4d7ada0

Change model-x to UrangDiffusion 1.1

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -18,7 +18,7 @@ pipe1 = StableDiffusionXLPipeline.from_pretrained(
18
  pipe1.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe1.scheduler.config)
19
 
20
  pipe2 = StableDiffusionXLPipeline.from_pretrained(
21
- "kayfahaarukku/model-x",
22
  torch_dtype=torch.float16,
23
  custom_pipeline="lpw_stable_diffusion_xl",
24
  )
@@ -34,7 +34,7 @@ def generate_comparison(prompt, negative_prompt, use_defaults, resolution, guida
34
  seed = random.randint(0, 99999999)
35
  if use_defaults:
36
  prompt = f"{prompt}, masterpiece, best quality, amazing quality, very aesthetic"
37
- negative_prompt = f"lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, displeasing, {negative_prompt}"
38
  generator = torch.manual_seed(seed)
39
 
40
  def callback(step, timestep, latents):
@@ -83,10 +83,10 @@ def interface_fn(prompt, negative_prompt, use_defaults, resolution, guidance_sca
83
  def reset_inputs():
84
  return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='896x1152'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True), gr.update(value='')
85
 
86
- with gr.Blocks(title="Model X Comparison Demo", theme="NoCrypt/miku@1.2.1") as demo:
87
  gr.HTML(
88
- "<h1>UrangDiffusion 1.0 vs Model X Comparison Demo</h1>"
89
- "This demo showcases a comparison between UrangDiffusion 1.0 and Model X."
90
  )
91
  with gr.Row():
92
  with gr.Column():
@@ -111,7 +111,7 @@ with gr.Blocks(title="Model X Comparison Demo", theme="NoCrypt/miku@1.2.1") as d
111
  with gr.Column():
112
  with gr.Row():
113
  output_image1 = gr.Image(type="pil", label="UrangDiffusion 1.0")
114
- output_image2 = gr.Image(type="pil", label="Model X")
115
  with gr.Accordion("Parameters", open=False):
116
  gr.Markdown(
117
  """
 
18
  pipe1.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe1.scheduler.config)
19
 
20
  pipe2 = StableDiffusionXLPipeline.from_pretrained(
21
+ "kayfahaarukku/UrangDiffusion-1.1",
22
  torch_dtype=torch.float16,
23
  custom_pipeline="lpw_stable_diffusion_xl",
24
  )
 
34
  seed = random.randint(0, 99999999)
35
  if use_defaults:
36
  prompt = f"{prompt}, masterpiece, best quality, amazing quality, very aesthetic"
37
+ negative_prompt = f"nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, displeasing, {negative_prompt}"
38
  generator = torch.manual_seed(seed)
39
 
40
  def callback(step, timestep, latents):
 
83
  def reset_inputs():
84
  return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='896x1152'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True), gr.update(value='')
85
 
86
+ with gr.Blocks(title="UrangDiffusion Comparison Demo", theme="NoCrypt/miku@1.2.1") as demo:
87
  gr.HTML(
88
+ "<h1>UrangDiffusion 1.0 vs 1.1 Comparison Demo</h1>"
89
+ "This demo showcases a comparison between UrangDiffusion 1.0 and 1.1."
90
  )
91
  with gr.Row():
92
  with gr.Column():
 
111
  with gr.Column():
112
  with gr.Row():
113
  output_image1 = gr.Image(type="pil", label="UrangDiffusion 1.0")
114
+ output_image2 = gr.Image(type="pil", label="UrangDiffusion 1.1")
115
  with gr.Accordion("Parameters", open=False):
116
  gr.Markdown(
117
  """