SALMONN-7B-gradio

Paused

fffiloni commited on Feb 13, 2024

Commit

b902051

verified ·

1 Parent(s): c289798

Update model_zero.py

Files changed (1) hide show

model_zero.py CHANGED Viewed

@@ -43,7 +43,8 @@ class SALMONN(nn.Module):
         lora_dropout=0.1,
         second_per_frame=0.333333,
         second_stride=0.333333,
-        low_resource=False
     ):
         super().__init__()

         lora_dropout=0.1,
         second_per_frame=0.333333,
         second_stride=0.333333,
+        low_resource=False,
+        device=None
     ):
         super().__init__()