Spaces:
Running
on
L40S
Running
on
L40S
Upload app_hg.py with huggingface_hub
Browse files
app_hg.py
CHANGED
@@ -21,6 +21,7 @@
|
|
21 |
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
22 |
import spaces
|
23 |
import os
|
|
|
24 |
import warnings
|
25 |
import argparse
|
26 |
import gradio as gr
|
|
|
21 |
# by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
|
22 |
import spaces
|
23 |
import os
|
24 |
+
os.environ['CUDA_HOME'] = '/usr/local/cuda'
|
25 |
import warnings
|
26 |
import argparse
|
27 |
import gradio as gr
|