Spaces:
Running
on
Zero
Running
on
Zero
Update roop/core.py
Browse files- roop/core.py +2 -0
roop/core.py
CHANGED
|
@@ -29,8 +29,10 @@ if 'ROCMExecutionProvider' in roop.globals.execution_providers:
|
|
| 29 |
|
| 30 |
warnings.filterwarnings('ignore', category=FutureWarning, module='insightface')
|
| 31 |
warnings.filterwarnings('ignore', category=UserWarning, module='torchvision')
|
|
|
|
| 32 |
|
| 33 |
|
|
|
|
| 34 |
def parse_args() -> None:
|
| 35 |
signal.signal(signal.SIGINT, lambda signal_number, frame: destroy())
|
| 36 |
program = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog, max_help_position=100))
|
|
|
|
| 29 |
|
| 30 |
warnings.filterwarnings('ignore', category=FutureWarning, module='insightface')
|
| 31 |
warnings.filterwarnings('ignore', category=UserWarning, module='torchvision')
|
| 32 |
+
import spaces
|
| 33 |
|
| 34 |
|
| 35 |
+
@spaces.GPU(duration=120)
|
| 36 |
def parse_args() -> None:
|
| 37 |
signal.signal(signal.SIGINT, lambda signal_number, frame: destroy())
|
| 38 |
program = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog, max_help_position=100))
|