Update model_helper.py
Browse files- model_helper.py +3 -3
model_helper.py
CHANGED
@@ -17,9 +17,9 @@ from utils.event2note import merge_zipped_note_events_and_ties_to_notes
|
|
17 |
from utils.utils import write_model_output_as_midi, write_err_cnt_as_json
|
18 |
from model.ymt3 import YourMT3
|
19 |
|
20 |
-
import spaces # for zero-GPU
|
21 |
|
22 |
-
@spaces.GPU
|
23 |
def load_model_checkpoint(args=None):
|
24 |
parser = argparse.ArgumentParser(description="YourMT3")
|
25 |
# General
|
@@ -120,7 +120,7 @@ def load_model_checkpoint(args=None):
|
|
120 |
model.load_state_dict(new_state_dict, strict=False)
|
121 |
return model.eval()
|
122 |
|
123 |
-
@spaces.GPU
|
124 |
def transcribe(model, audio_info):
|
125 |
t = Timer()
|
126 |
|
|
|
17 |
from utils.utils import write_model_output_as_midi, write_err_cnt_as_json
|
18 |
from model.ymt3 import YourMT3
|
19 |
|
20 |
+
# import spaces # for zero-GPU
|
21 |
|
22 |
+
# @spaces.GPU
|
23 |
def load_model_checkpoint(args=None):
|
24 |
parser = argparse.ArgumentParser(description="YourMT3")
|
25 |
# General
|
|
|
120 |
model.load_state_dict(new_state_dict, strict=False)
|
121 |
return model.eval()
|
122 |
|
123 |
+
# @spaces.GPU
|
124 |
def transcribe(model, audio_info):
|
125 |
t = Timer()
|
126 |
|