Hugging Face's logo

Spaces:
Manjushri
/
Manju-Dream-Booth-GPU
Running on T4

App Files
Manjushri commited on
Commit
9e995ef
·
verified ·
1 Parent(s): b126be6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ torch.cuda.empty_cache()
12
 
13
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo", torch_dtype=torch.bfloat16)
14
  pipe = pipe.to(device)
 
15
 
16
 
17
  def genie (Prompt, height, width, seed):
 
12
 
13
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo", torch_dtype=torch.bfloat16)
14
  pipe = pipe.to(device)
15
+ pipe.enable_model_cpu_offload()
16
 
17
 
18
  def genie (Prompt, height, width, seed):