Spaces:
Running
on
Zero
Running
on
Zero
Update sam_audio/model/config.py
Browse files
sam_audio/model/config.py
CHANGED
|
@@ -156,7 +156,7 @@ class ClapRankerConfig(RankerConfig):
|
|
| 156 |
class JudgeRankerConfig(RankerConfig):
|
| 157 |
kind: str = "judge"
|
| 158 |
|
| 159 |
-
def __init__(self, checkpoint_or_model_id: str = "
|
| 160 |
self.checkpoint_or_model_id = checkpoint_or_model_id
|
| 161 |
|
| 162 |
|
|
|
|
| 156 |
class JudgeRankerConfig(RankerConfig):
|
| 157 |
kind: str = "judge"
|
| 158 |
|
| 159 |
+
def __init__(self, checkpoint_or_model_id: str = "facebook/sam-audio-judge"):
|
| 160 |
self.checkpoint_or_model_id = checkpoint_or_model_id
|
| 161 |
|
| 162 |
|