ford442 commited on
Commit
aa3e2b2
·
verified ·
1 Parent(s): ab983f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -62,7 +62,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained("ford442/stable-diffusion-3.5-me
62
 
63
  #pipe.scheduler.config.requires_aesthetics_score = False
64
  #pipe.enable_model_cpu_offload()
65
- pipe.to(device)
66
  #pipe = torch.compile(pipe)
67
  # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
68
 
@@ -264,7 +264,7 @@ with gr.Blocks(css=css) as demo:
264
  minimum=0.0,
265
  maximum=10.0,
266
  step=0.1,
267
- value=5.0, # Replace with defaults that work for your model
268
  )
269
  num_inference_steps = gr.Slider(
270
  label="Number of inference steps",
 
62
 
63
  #pipe.scheduler.config.requires_aesthetics_score = False
64
  #pipe.enable_model_cpu_offload()
65
+ pipe.to(torch_device="cuda", torch_dtype=torch.bfloat16)
66
  #pipe = torch.compile(pipe)
67
  # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
68
 
 
264
  minimum=0.0,
265
  maximum=10.0,
266
  step=0.1,
267
+ value=4.2, # Replace with defaults that work for your model
268
  )
269
  num_inference_steps = gr.Slider(
270
  label="Number of inference steps",