skroed commited on
Commit
ce29d3b
1 Parent(s): ff2924b

Fix: do not use torch compile for compat.

Browse files
Files changed (1) hide show
  1. handler.py +0 -3
handler.py CHANGED
@@ -11,9 +11,6 @@ class EndpointHandler:
11
  "cvssp/audioldm2-music", torch_dtype=torch.float16
12
  )
13
  self.pipeline.to("cuda")
14
- self.pipeline.unet = torch.compile(
15
- self.pipeline.unet, mode="reduce-overhead", fullgraph=True
16
- )
17
  self.pipeline.scheduler = DPMSolverMultistepScheduler.from_config(
18
  self.pipeline.scheduler.config
19
  )
 
11
  "cvssp/audioldm2-music", torch_dtype=torch.float16
12
  )
13
  self.pipeline.to("cuda")
 
 
 
14
  self.pipeline.scheduler = DPMSolverMultistepScheduler.from_config(
15
  self.pipeline.scheduler.config
16
  )