[Admin maintenance] Support new ZeroGPU hardware

Thank you so much for having shared this Space with the community on this demo. We have upgraded the ZeroGPU infra-structure to run on modern blackwell architecture.
For that, we need to upgrade your demo to support that. This PR fixes your demo to work with the new architecture. As this is something we broke on our end, we may merge this PR autonomously. If this breaks unexpectedly or brings unintended consequences, feel free to revert, modify or otherwise. Any issues you can email apolinario@huggingface.co

README.md

@@ -4,8 +4,8 @@ emoji: 📸
 colorFrom: indigo
 colorTo: gray
 sdk: gradio
-sdk_version: 4.38.1
+sdk_version: 5.49.1
 app_file: app.py
 pinned: false
-python_version: 3.1
+python_version: "3.10"
 ---

app.py

@@ -2,12 +2,24 @@ import datetime
 import json
 import os
 
+import spaces
 import gradio as gr
 from huggingface_hub import hf_hub_download
-import spaces
 import PIL.Image
 import numpy as np
 import torch
+
+# Force xformers to use Cutlass op (FA3 Hopper kernel crashes on Blackwell sm_120).
+import xformers
+import xformers.ops as _xops
+_orig_mea = _xops.memory_efficient_attention
+_cutlass_ops = (_xops.fmha.cutlass.FwOp, _xops.fmha.cutlass.BwOp)
+def _mea_cutlass(*a, **kw):
+    if kw.get("op") is None:
+        kw["op"] = _cutlass_ops
+    return _orig_mea(*a, **kw)
+_xops.memory_efficient_attention = _mea_cutlass
+xformers.ops.memory_efficient_attention = _mea_cutlass
 import torchvision.transforms.functional
 from numpy import deg2rad
 from omegaconf import OmegaConf

requirements.txt

@@ -1,18 +1,17 @@
-torch==2.1.2
-pytorch_lightning==2.1.2
-deepspeed==0.15.0
+torch==2.8.0
+pytorch_lightning
 taming-transformers==0.0.1
 diffusers==0.30.1
 transformers
-torchvision==0.16.2
+torchvision==0.23.0
 scipy==1.14.1
 einops==0.8.0
 kornia==0.7.3
 open_clip_torch==2.20.0
 openai-clip==1.0.1
 numpy==1.26.3
-xformers==0.0.23.post1
+xformers
 timm==1.0.9
 av==12.3.0
-gradio==5.8.0
-huggingface_hub
+huggingface_hub
+