John6666 commited on
Commit
4781298
·
verified ·
1 Parent(s): 3f860f3

Upload handler.py

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