mimbres commited on
Commit
0a4ad71
1 Parent(s): 23c3720

Update model_helper.py

Browse files
Files changed (1) hide show
  1. model_helper.py +3 -1
model_helper.py CHANGED
@@ -17,7 +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
 
 
21
  def load_model_checkpoint(args=None):
22
  parser = argparse.ArgumentParser(description="YourMT3")
23
  # General
@@ -118,7 +120,7 @@ def load_model_checkpoint(args=None):
118
  model.load_state_dict(new_state_dict, strict=False)
119
  return model.eval()
120
 
121
-
122
  def transcribe(model, audio_info):
123
  t = Timer()
124
 
 
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