Fix Space for new Blackwell GPUs

#16
by cbensimon HF Staff - opened
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ pipe = Flux2Pipeline.from_pretrained(
79
  pipe.to(device)
80
 
81
  # Pull pre-compiled Flux2 Transformer blocks from HF hub
82
- spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2", variant="fa3")
83
 
84
  def image_to_data_uri(img):
85
  buffered = io.BytesIO()
 
79
  pipe.to(device)
80
 
81
  # Pull pre-compiled Flux2 Transformer blocks from HF hub
82
+ spaces.aoti_blocks_load(pipe.transformer, "zerogpu-aoti/FLUX.2")
83
 
84
  def image_to_data_uri(img):
85
  buffered = io.BytesIO()