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(
|