debasishmohanty commited on
Commit
f33e451
·
1 Parent(s): f8c53d2

modified config and handler to get the runner path correct

Browse files
Files changed (2) hide show
  1. handler.py +5 -5
  2. or/fastpitch/config.json +2 -2
handler.py CHANGED
@@ -115,12 +115,12 @@ class EndpointHandler():
115
  print(f"cuda version>> ",{torch.cuda.get_device_capability(0)[0]})
116
  models = {}
117
  odia_model = Synthesizer(
118
- tts_checkpoint="./or/fastpitch/best_model.pth",
119
- tts_config_path="./or/fastpitch/config.json",
120
  # modify this config.json to proper model path
121
- tts_speakers_file="./or/fastpitch/speakers.pth",
122
- vocoder_checkpoint="./or/hifigan/best_model.pth",
123
- vocoder_config="./or/hifigan/config.json",
124
  use_cuda = True
125
  )
126
  models["or"] = odia_model
 
115
  print(f"cuda version>> ",{torch.cuda.get_device_capability(0)[0]})
116
  models = {}
117
  odia_model = Synthesizer(
118
+ tts_checkpoint="/repository/or/fastpitch/best_model.pth",
119
+ tts_config_path="/repository/or/fastpitch/config.json",
120
  # modify this config.json to proper model path
121
+ tts_speakers_file="/repository/or/fastpitch/speakers.pth",
122
+ vocoder_checkpoint="/repository/or/hifigan/best_model.pth",
123
+ vocoder_config="/repository/or/hifigan/config.json",
124
  use_cuda = True
125
  )
126
  models["or"] = odia_model
or/fastpitch/config.json CHANGED
@@ -178,7 +178,7 @@
178
  "max_duration": 75,
179
  "num_speakers": 2,
180
  "use_speaker_embedding": true,
181
- "speakers_file": "./or/fastpitch/speakers.pth",
182
  "use_d_vector_file": false,
183
  "d_vector_dim": 512,
184
  "d_vector_file": null,
@@ -191,7 +191,7 @@
191
  },
192
  "return_wav": false,
193
  "num_speakers": 2,
194
- "speakers_file": "./or/fastpitch/speakers.pth",
195
  "use_speaker_embedding": true,
196
  "use_d_vector_file": false,
197
  "d_vector_file": "",
 
178
  "max_duration": 75,
179
  "num_speakers": 2,
180
  "use_speaker_embedding": true,
181
+ "speakers_file": "/repository/or/fastpitch/speakers.pth",
182
  "use_d_vector_file": false,
183
  "d_vector_dim": 512,
184
  "d_vector_file": null,
 
191
  },
192
  "return_wav": false,
193
  "num_speakers": 2,
194
+ "speakers_file": "/repository/or/fastpitch/speakers.pth",
195
  "use_speaker_embedding": true,
196
  "use_d_vector_file": false,
197
  "d_vector_file": "",