Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -195,6 +195,7 @@ else:
|
|
| 195 |
or "M4" in gpu_name.upper()
|
| 196 |
or "T4" in gpu_name.upper()
|
| 197 |
or "TITAN" in gpu_name.upper()
|
|
|
|
| 198 |
): # A10#A100#V100#A40#P40#M40#K80#A4500
|
| 199 |
if_gpu_ok = True # 至少有一张能用的N卡
|
| 200 |
gpu_infos.append("%s\t%s" % (i, gpu_name))
|
|
@@ -261,7 +262,7 @@ for root, dirs, files in os.walk(index_root, topdown=False):
|
|
| 261 |
index_paths.append("%s/%s" % (root, name))
|
| 262 |
|
| 263 |
|
| 264 |
-
|
| 265 |
def vc_single(
|
| 266 |
sid,
|
| 267 |
input_audio_path,
|
|
|
|
| 195 |
or "M4" in gpu_name.upper()
|
| 196 |
or "T4" in gpu_name.upper()
|
| 197 |
or "TITAN" in gpu_name.upper()
|
| 198 |
+
or "ZERO" in gpu_name.upper()
|
| 199 |
): # A10#A100#V100#A40#P40#M40#K80#A4500
|
| 200 |
if_gpu_ok = True # 至少有一张能用的N卡
|
| 201 |
gpu_infos.append("%s\t%s" % (i, gpu_name))
|
|
|
|
| 262 |
index_paths.append("%s/%s" % (root, name))
|
| 263 |
|
| 264 |
|
| 265 |
+
@spaces.GPU
|
| 266 |
def vc_single(
|
| 267 |
sid,
|
| 268 |
input_audio_path,
|