zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Remove torch from requirements

Browse Source
This commit is contained in:
oobabooga
2023-04-06 20:31:04 -03:00
parent e80d0b0160
commit de7fb66d52
2 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/silero_tts/requirements.txt
View File

@@ -3,5 +3,3 @@ num2words
omegaconf omegaconf
pydub pydub
PyYAML PyYAML
torch
torchaudio

4
extensions/silero_tts/test_tts.py
View File

@@ -34,18 +34,15 @@ table = str.maketrans({
'"': """, '"': """,
}) })
def xmlesc(txt): def xmlesc(txt):
return txt.translate(table) return txt.translate(table)
def load_model(): def load_model():
model, example_text = torch.hub.load(repo_or_dir='snakers4/silero-models', model='silero_tts', language=params['language'], speaker=params['model_id']) model, example_text = torch.hub.load(repo_or_dir='snakers4/silero-models', model='silero_tts', language=params['language'], speaker=params['model_id'])
model.to(params['device']) model.to(params['device'])
return model return model
model = load_model() model = load_model()
def output_modifier(string): def output_modifier(string):
""" """
This function is applied to the model outputs. This function is applied to the model outputs.
@@ -73,7 +70,6 @@ def output_modifier(string):
print(string) print(string)
if __name__ == '__main__': if __name__ == '__main__':
import sys import sys
output_modifier(sys.argv[1]) output_modifier(sys.argv[1])