ford442 commited on
Commit
10268d7
·
verified ·
1 Parent(s): 79ebb83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -62,7 +62,8 @@ 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=device, dtype=torch.bfloat16)
 
66
  #pipe = torch.compile(pipe)
67
  # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
68
 
 
62
 
63
  #pipe.scheduler.config.requires_aesthetics_score = False
64
  #pipe.enable_model_cpu_offload()
65
+ pipe.to(device)
66
+ #pipe.to(device=device, dtype=torch.bfloat16)
67
  #pipe = torch.compile(pipe)
68
  # pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear")
69