Serhiy Stetskovych commited on
Commit
8fc33c0
·
1 Parent(s): a257548

Move single model

Browse files
Files changed (2) hide show
  1. .gitattributes +0 -1
  2. infer.py +2 -1
.gitattributes CHANGED
@@ -35,7 +35,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  weights/plbert.bin filter=lfs diff=lfs merge=lfs -text
37
  weights/asr.bin filter=lfs diff=lfs merge=lfs -text
38
- weights/filatov.bin filter=lfs diff=lfs merge=lfs -text
39
  weights/jdc.bin filter=lfs diff=lfs merge=lfs -text
40
  voices/Олег[[:space:]]Лепенець.wav filter=lfs diff=lfs merge=lfs -text
41
  voices/Слава[[:space:]]Красовська.wav filter=lfs diff=lfs merge=lfs -text
 
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  weights/plbert.bin filter=lfs diff=lfs merge=lfs -text
37
  weights/asr.bin filter=lfs diff=lfs merge=lfs -text
 
38
  weights/jdc.bin filter=lfs diff=lfs merge=lfs -text
39
  voices/Олег[[:space:]]Лепенець.wav filter=lfs diff=lfs merge=lfs -text
40
  voices/Слава[[:space:]]Красовська.wav filter=lfs diff=lfs merge=lfs -text
infer.py CHANGED
@@ -85,7 +85,8 @@ multi_path = hf_hub_download(repo_id='patriotyk/styletts2_ukrainian_multispeaker
85
  params_multi = torch.load(multi_path, map_location='cpu')
86
 
87
 
88
- params_single = torch.load('weights/filatov.bin', map_location='cpu')
 
89
 
90
 
91
  load_state_dict(model_single, params_single)
 
85
  params_multi = torch.load(multi_path, map_location='cpu')
86
 
87
 
88
+ single_path = hf_hub_download(repo_id='patriotyk/styletts2_ukrainian_single', filename="pytorch_model.bin")
89
+ params_single = torch.load(single_path, map_location='cpu')
90
 
91
 
92
  load_state_dict(model_single, params_single)