multimodalart HF Staff cbensimon HF Staff commited on
Commit
4593e14
·
1 Parent(s): 497aa59

Fix Space for new Blackwell GPUs (#16)

Browse files

- Fix Space for new Blackwell GPUs (fe4fe480c7ade15ea7c89df8cf7f6d7b7a4dace2)


Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>

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()