Spaces:
Running
on
A100
Running
on
A100
base model test ok
Browse files
acestep/acestep_v15_pipeline.py
CHANGED
|
@@ -9,8 +9,6 @@ import sys
|
|
| 9 |
for proxy_var in ['http_proxy', 'https_proxy', 'HTTP_PROXY', 'HTTPS_PROXY', 'ALL_PROXY']:
|
| 10 |
os.environ.pop(proxy_var, None)
|
| 11 |
|
| 12 |
-
os.environ['CUDA_VISIBLE_DEVICES'] = '7' # Adjust as needed
|
| 13 |
-
|
| 14 |
from .handler import AceStepHandler
|
| 15 |
from .gradio_ui import create_gradio_interface
|
| 16 |
|
|
|
|
| 9 |
for proxy_var in ['http_proxy', 'https_proxy', 'HTTP_PROXY', 'HTTPS_PROXY', 'ALL_PROXY']:
|
| 10 |
os.environ.pop(proxy_var, None)
|
| 11 |
|
|
|
|
|
|
|
| 12 |
from .handler import AceStepHandler
|
| 13 |
from .gradio_ui import create_gradio_interface
|
| 14 |
|