drod75 commited on
Commit
326f7fa
1 Parent(s): 2654b52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,8 +9,8 @@ from diffusers import DiffusionPipeline
9
  import torch
10
 
11
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
12
- pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
13
  pipe.enable_model_cpu_offload()
 
14
 
15
  prompt = st.text_input('Enter a prompt for your Anime character: ')
16
 
 
9
  import torch
10
 
11
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
 
12
  pipe.enable_model_cpu_offload()
13
+ pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
14
 
15
  prompt = st.text_input('Enter a prompt for your Anime character: ')
16