submission-template

Sleeping

seronk commited on Jan 31

Commit

9ce63fb

verified ·

1 Parent(s): 99568ae

Update tasks/audio.py

Files changed (1) hide show

tasks/audio.py CHANGED Viewed

@@ -6,8 +6,7 @@ import random
 import os
 import torch
 from torch.utils.data import DataLoader
-from Model_Loader import load_model
 from .utils.evaluation import AudioEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info

 import os
 import torch
 from torch.utils.data import DataLoader
+from Model_Loader import M5, load_model
 from .utils.evaluation import AudioEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info