Hugging Face's logo

Spaces:
bghira
/
pseudo-flex-v2
Running on Zero

App Files Community
PseudoTerminal X commited on
Commit
35425ea
·
verified ·
1 Parent(s): d788f85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from diffusers import DiffusionPipeline
3
  import gradio as gr
4
 
5
  pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2', torch_dtype=torch.float16)
6
- pipe.unet = torch.compile(pipe.unet)
7
  pipe.to('cuda')
8
 
9
  @spaces.GPU
 
3
  import gradio as gr
4
 
5
  pipe = DiffusionPipeline.from_pretrained('ptx0/terminus-xl-velocity-v2', torch_dtype=torch.float16)
6
+ #pipe.unet = torch.compile(pipe.unet)
7
  pipe.to('cuda')
8
 
9
  @spaces.GPU