R-Kentaren commited on
Commit
c330742
ยท
verified ยท
1 Parent(s): 98ddd74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -126,7 +126,8 @@ def model_data(model_name):
126
  def load_hubert():
127
  from fairseq import fairseq
128
 
129
- fairseq.GradMultiply.forward = forward_dml
 
130
 
131
  models, _, _ = fairseq.load_model(
132
  f"{BASE_DIR}/hubert_base.pt",
 
126
  def load_hubert():
127
  from fairseq import fairseq
128
 
129
+ forward_dml = fairseq.GradMultiply.forward
130
+
131
 
132
  models, _, _ = fairseq.load_model(
133
  f"{BASE_DIR}/hubert_base.pt",