Hugging Face's logo

Spaces:
fffiloni
/
L4GM-demo
Paused

App Files Community
fffiloni commited on
Commit
aa3cb1c
·
verified ·
1 Parent(s): 93105c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from glob import glob
11
  env_list = os.environ['PATH'].split(':')
12
  env_list.append('/usr/local/cuda/bin')
13
  os.environ['PATH'] = ':'.join(env_list)
14
- os.environ['TORCH_CUDA_ARCH_LIST'] = '8.9'
15
 
16
  import gradio as gr
17
  from huggingface_hub import snapshot_download
 
11
  env_list = os.environ['PATH'].split(':')
12
  env_list.append('/usr/local/cuda/bin')
13
  os.environ['PATH'] = ':'.join(env_list)
14
+ os.environ['TORCH_CUDA_ARCH_LIST'] = '8.6'
15
 
16
  import gradio as gr
17
  from huggingface_hub import snapshot_download