Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

python compat update for contextlib

This commit is contained in:
erogol 2020-11-06 13:34:11 +01:00
parent c80225544e
commit ea976b0543
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/utils/generic_utils.py
View File

@ -3,7 +3,7 @@ import glob
import os import os
import shutil import shutil
import subprocess import subprocess
from contextlib import nullcontext import contextlib
import torch import torch
@ -11,6 +11,9 @@ import torch
def set_amp_context(mixed_precision): def set_amp_context(mixed_precision):
if mixed_precision: if mixed_precision:
cm = torch.cuda.amp.autocast() cm = torch.cuda.amp.autocast()
else:
if os.python.version<=3.6:
cm = contextlib.suppress()
else: else:
cm = nullcontext() cm = nullcontext()
return cm return cm