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

Update model_helper.py

Browse files
Files changed (1) hide show
  1. 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