John6666 commited on
Commit
3dd1afc
·
verified ·
1 Parent(s): bacfd7c

Upload handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -17,9 +17,9 @@ def compile_pipeline(pipe) -> Any:
17
  pipe.transformer.fuse_qkv_projections()
18
  pipe.transformer.to(memory_format=torch.channels_last)
19
  pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=False, dynamic=False, backend="inductor")
20
- pipe.vae.fuse_qkv_projections()
21
- pipe.vae.to(memory_format=torch.channels_last)
22
- pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=False, dynamic=False, backend="inductor")
23
  return pipe
24
 
25
  class EndpointHandler:
 
17
  pipe.transformer.fuse_qkv_projections()
18
  pipe.transformer.to(memory_format=torch.channels_last)
19
  pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=False, dynamic=False, backend="inductor")
20
+ #pipe.vae.fuse_qkv_projections()
21
+ #pipe.vae.to(memory_format=torch.channels_last)
22
+ #pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=False, dynamic=False, backend="inductor")
23
  return pipe
24
 
25
  class EndpointHandler: