mimbres commited on
Commit
8f5b8ef
1 Parent(s): e6eed3e

Update model_helper.py

Browse files
Files changed (1) hide show
  1. model_helper.py +3 -3
model_helper.py CHANGED
@@ -1,5 +1,5 @@
1
  # @title Model helper
2
- import spaces # for zero-GPU
3
 
4
  import os
5
  from collections import Counter
@@ -21,7 +21,7 @@ from model.ymt3 import YourMT3
21
 
22
 
23
 
24
- @spaces.GPU
25
  def load_model_checkpoint(args=None):
26
  parser = argparse.ArgumentParser(description="YourMT3")
27
  # General
@@ -122,7 +122,7 @@ def load_model_checkpoint(args=None):
122
  model.load_state_dict(new_state_dict, strict=False)
123
  return model.eval()
124
 
125
- @spaces.GPU
126
  def transcribe(model, audio_info):
127
  t = Timer()
128
 
 
1
  # @title Model helper
2
+ # import spaces # for zero-GPU
3
 
4
  import os
5
  from collections import Counter
 
21
 
22
 
23
 
24
+
25
  def load_model_checkpoint(args=None):
26
  parser = argparse.ArgumentParser(description="YourMT3")
27
  # General
 
122
  model.load_state_dict(new_state_dict, strict=False)
123
  return model.eval()
124
 
125
+
126
  def transcribe(model, audio_info):
127
  t = Timer()
128