jbilcke-hf HF staff commited on
Commit
c3babc6
·
verified ·
1 Parent(s): 10002d2

Update handler.py

Browse files
Files changed (1) hide show
  1. 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
- output_format="mp4",
154
- output_codec="h264",
155
- output_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..
 
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..