PseudoTerminal X commited on
Commit
68dc9bc
1 Parent(s): d17e9f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import spaces, torch
2
  from diffusers import DiffusionPipeline
3
  import gradio as gr
4
 
5
- pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2', torch_dtype=torch.float16)
6
  #pipe.unet = torch.compile(pipe.unet)
7
  pipe.to('cuda')
8
 
@@ -10,9 +10,10 @@ pipe.to('cuda')
10
  def generate(prompt):
11
  return pipe(
12
  prompt,
 
13
  guidance_rescale=0.7,
14
- guidance_scale=11.5,
15
- num_inference_steps=25,
16
  ).images
17
 
18
  gr.Interface(
 
2
  from diffusers import DiffusionPipeline
3
  import gradio as gr
4
 
5
+ pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2', torch_dtype=torch.bfloat16)
6
  #pipe.unet = torch.compile(pipe.unet)
7
  pipe.to('cuda')
8
 
 
10
  def generate(prompt):
11
  return pipe(
12
  prompt,
13
+ negative_prompt="underexposed, blurry, ugly, washed-out",
14
  guidance_rescale=0.7,
15
+ guidance_scale=9.5,
16
+ num_inference_steps=30,
17
  ).images
18
 
19
  gr.Interface(