mirror of https://github.com/coqui-ai/TTS.git
import condition update for synthesis with TF
This commit is contained in:
parent
84c5c4a587
commit
68dbcee746
utils
|
@ -1,6 +1,6 @@
|
||||||
import pkg_resources
|
import pkg_resources
|
||||||
installed = {pkg.key for pkg in pkg_resources.working_set}
|
installed = {pkg.key for pkg in pkg_resources.working_set}
|
||||||
if 'tensorflow' in installed:
|
if 'tensorflow' in installed or 'tensorflow-gpu' in installed:
|
||||||
import tensorflow as tf
|
import tensorflow as tf
|
||||||
import torch
|
import torch
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
Loading…
Reference in New Issue