Update handler.py
Browse files- handler.py +3 -3
handler.py
CHANGED
@@ -150,9 +150,9 @@ class EndpointHandler:
|
|
150 |
# Initialize Varnish for post-processing
|
151 |
self.varnish = Varnish(
|
152 |
device="cuda" if torch.cuda.is_available() else "cpu",
|
153 |
-
|
154 |
-
|
155 |
-
|
156 |
model_base_dir="/repository/varnish",
|
157 |
|
158 |
# there is currently a bug with MMAudio and/or torch and/or the weight format and/or version..
|
|
|
150 |
# Initialize Varnish for post-processing
|
151 |
self.varnish = Varnish(
|
152 |
device="cuda" if torch.cuda.is_available() else "cpu",
|
153 |
+
format="mp4",
|
154 |
+
codec="h264",
|
155 |
+
quality=17,
|
156 |
model_base_dir="/repository/varnish",
|
157 |
|
158 |
# there is currently a bug with MMAudio and/or torch and/or the weight format and/or version..
|