Преглед изворни кода

added missing nested-modules-discovery from global entrypoint like torch.hub

Krzysztof Kudrynski пре 6 година
родитељ
комит
14fe91fad5
1 измењених фајлова са 4 додато и 0 уклоњено
  1. 4 0
      PyTorch/SpeechSynthesis/Tacotron2/models.py

+ 4 - 0
PyTorch/SpeechSynthesis/Tacotron2/models.py

@@ -25,6 +25,10 @@
 #
 # *****************************************************************************
 
+import sys
+from os.path import abspath, dirname
+# enabling modules discovery from global entrypoint
+sys.path.append(abspath(dirname(__file__)+'/'))
 from tacotron2.model import Tacotron2
 from waveglow.model import WaveGlow
 import torch