Spaces:
Running
on
Zero
Running
on
Zero
Update sam_audio/model/config.py
Browse files
sam_audio/model/config.py
CHANGED
|
@@ -213,7 +213,7 @@ class SAMAudioConfig:
|
|
| 213 |
anchor_embedding_dim: int = 128,
|
| 214 |
visual_ranker=None,
|
| 215 |
text_ranker=None,
|
| 216 |
-
span_predictor
|
| 217 |
):
|
| 218 |
self.in_channels = in_channels
|
| 219 |
self.audio_codec = DACVAEConfig(**(audio_codec or {}))
|
|
|
|
| 213 |
anchor_embedding_dim: int = 128,
|
| 214 |
visual_ranker=None,
|
| 215 |
text_ranker=None,
|
| 216 |
+
span_predictor: Optional[str] = "pe-a-frame-large",
|
| 217 |
):
|
| 218 |
self.in_channels = in_channels
|
| 219 |
self.audio_codec = DACVAEConfig(**(audio_codec or {}))
|