Manjushri commited on
Commit
97ddc59
·
verified ·
1 Parent(s): 624960b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
40
  torch.cuda.empty_cache()
41
  model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
42
  pipe = DiffusionPipeline.from_pretrained(model_repo_id)
43
- #pipe = pipe.to(device)
44
  pipe.enable_model_cpu_offload()
45
  pipe.vae.enable_slicing()
46
  pipe.vae.enable_tiling()
 
40
  torch.cuda.empty_cache()
41
  model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
42
  pipe = DiffusionPipeline.from_pretrained(model_repo_id)
43
+ pipe = pipe.to(device)
44
  pipe.enable_model_cpu_offload()
45
  pipe.vae.enable_slicing()
46
  pipe.vae.enable_tiling()