Commit
•
91a9d80
1
Parent(s):
ccbe184
Update handler.py
Browse files- 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",
|
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=
|
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(
|