VoiceCloning-be commited on
Commit
f658b84
1 Parent(s): 3e8e7fc

Update core.py

Browse files
Files changed (1) hide show
  1. core.py +4 -0
core.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import sys
3
  import json
@@ -363,6 +364,7 @@ def run_tts_script(
363
 
364
 
365
  # Preprocess
 
366
  def run_preprocess_script(
367
  model_name: str,
368
  dataset_path: str,
@@ -395,6 +397,7 @@ def run_preprocess_script(
395
 
396
 
397
  # Extract
 
398
  def run_extract_script(
399
  model_name: str,
400
  rvc_version: str,
@@ -437,6 +440,7 @@ def run_extract_script(
437
 
438
 
439
  # Train
 
440
  def run_train_script(
441
  model_name: str,
442
  rvc_version: str,
 
1
+ import spaces
2
  import os
3
  import sys
4
  import json
 
364
 
365
 
366
  # Preprocess
367
+ @spaces.GPU
368
  def run_preprocess_script(
369
  model_name: str,
370
  dataset_path: str,
 
397
 
398
 
399
  # Extract
400
+ @spaces.GPU
401
  def run_extract_script(
402
  model_name: str,
403
  rvc_version: str,
 
440
 
441
 
442
  # Train
443
+ @spaces.GPU
444
  def run_train_script(
445
  model_name: str,
446
  rvc_version: str,