jbilcke-hf HF staff commited on
Commit
91a9d80
1 Parent(s): ccbe184

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -254,7 +254,7 @@ class EndpointHandler:
254
 
255
  # Initialize Varnish for post-processing
256
  self.varnish = Varnish(
257
- device="cuda", # if torch.cuda.is_available() else "cpu",
258
  model_base_dir="/repository/varnish",
259
 
260
  # there is currently a bug with MMAudio and/or torch and/or the weight format and/or version..
@@ -273,7 +273,7 @@ class EndpointHandler:
273
  # magic_number = pickle_module.load(f, **pickle_load_args)
274
  # ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
275
  # _pickle.UnpicklingError: invalid load key, '<'.
276
- enable_mmaudio=False,
277
  )
278
 
279
  async def process_frames(
 
254
 
255
  # Initialize Varnish for post-processing
256
  self.varnish = Varnish(
257
+ device="cuda",
258
  model_base_dir="/repository/varnish",
259
 
260
  # there is currently a bug with MMAudio and/or torch and/or the weight format and/or version..
 
273
  # magic_number = pickle_module.load(f, **pickle_load_args)
274
  # ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
275
  # _pickle.UnpicklingError: invalid load key, '<'.
276
+ enable_mmaudio=True,
277
  )
278
 
279
  async def process_frames(