andrew3d commited on
Commit
95401cf
Β·
verified Β·
1 Parent(s): b3dedc5

Update app.py: set ATTN_BACKEND to sdpa

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -30,6 +30,9 @@
30
  import gradio as gr
31
  import os
32
  os.environ['SPCONV_ALGO'] = 'native'
 
 
 
33
  from typing import *
34
  import torch
35
  import numpy as np
 
30
  import gradio as gr
31
  import os
32
  os.environ['SPCONV_ALGO'] = 'native'
33
+ # Hi3DGen uses an attention backend that defaults to 'xformers', which requires an extra
34
+ # dependency not installed in this Space. Override the backend to use 'sdpa' instead.
35
+ os.environ['ATTN_BACKEND'] = 'sdpa'
36
  from typing import *
37
  import torch
38
  import numpy as np