Spaces:

fffiloni
/

L4GM-demo

Sleeping

fffiloni commited on 10 days ago

Commit

aa3cb1c

•

1 Parent(s): 93105c6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from glob import glob
 env_list = os.environ['PATH'].split(':')
 env_list.append('/usr/local/cuda/bin')
 os.environ['PATH'] = ':'.join(env_list)
-os.environ['TORCH_CUDA_ARCH_LIST'] = '8.9'
 import gradio as gr
 from huggingface_hub import snapshot_download

 env_list = os.environ['PATH'].split(':')
 env_list.append('/usr/local/cuda/bin')
 os.environ['PATH'] = ':'.join(env_list)
+os.environ['TORCH_CUDA_ARCH_LIST'] = '8.6'
 import gradio as gr
 from huggingface_hub import snapshot_download