ray-006 commited on
Commit
870a7f0
·
verified ·
1 Parent(s): 11f62b8

Update eval/metrics/judge.py

Browse files
Files changed (1) hide show
  1. eval/metrics/judge.py +1 -1
eval/metrics/judge.py CHANGED
@@ -10,7 +10,7 @@ from sam_audio import SAMAudioJudgeModel, SAMAudioJudgeProcessor
10
  class Judge(torch.nn.Module):
11
  def __init__(
12
  self,
13
- checkpoint: str = "facebook/sam-audio-judge",
14
  device: Optional[torch.device] = None,
15
  ):
16
  super().__init__()
 
10
  class Judge(torch.nn.Module):
11
  def __init__(
12
  self,
13
+ checkpoint: str = "ray-006/sample-audio-judge",
14
  device: Optional[torch.device] = None,
15
  ):
16
  super().__init__()