Spaces:
Running
Running
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 |
|