test

.gradio/certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

app.py

@@ -19,7 +19,6 @@ import numpy as np
 import gradio as gr
 import spaces
 
-
 # ── Lazy imports so Gradio loads even before heavy deps are installed ──────────
 
 def _import_deps():
@@ -106,15 +105,18 @@ def _get_open_data(ekd, ekr, date, param, levelist=[], **kwargs):
 @spaces.GPU
 def run_forecast(lead_time: int, num_chunks: int, progress=gr.Progress(track_tqdm=True)):
     """Download ICs, run AIFS, return status message."""
+
     global _runner, _states
 
-    torch, ekd, ekr, SimpleRunner, OpendataClient, plt, ccrs, cfeature, tri = _import_deps()
 
+    torch, ekd, ekr, SimpleRunner, OpendataClient, plt, ccrs, cfeature, tri = _import_deps()
+    print("1")
     # GPU check
     if not torch.cuda.is_available():
         return "❌ No CUDA GPU detected. This model requires an Ampere GPU (e.g. Colab L4/A100).", gr.update(choices=[])
 
     major, _ = torch.cuda.get_device_capability(0)
+    print("2")
     if major < 8:
         gpu = torch.cuda.get_device_name(0)
         return f"❌ GPU '{gpu}' is below Ampere (compute capability < 8.0). FlashAttention requires Ampere or newer.", gr.update(choices=[])
@@ -171,7 +173,12 @@ def run_forecast(lead_time: int, num_chunks: int, progress=gr.Progress(track_tqd
     # Load runner (cache it)
     yield "🤖  Loading AIFS Single v2 checkpoint from Hugging Face…", gr.update(choices=[])
     if _runner is None:
-        _runner = SimpleRunner(CHECKPOINT)
+        #_runner = SimpleRunner(CHECKPOINT)
+        # Override the attention implementation to avoid flash-attn
+        _runner = SimpleRunner(
+            CHECKPOINT,
+            overrides={"model.processor.attention_implementation": "scaled_dot_product_attention"}
+        )
 
     # Run inference
     _states = []

requirements.txt

@@ -1,15 +1,12 @@
-# spaces must be installed for ZeroGPU
 spaces
 
-# torch first — flash-attn depends on it at build time
 torch==2.11.0
 torch-geometric==2.6.1
 
 # flash-attn pre-built wheel matching the HF Space environment exactly:
 # Python 3.12 | CUDA 13.0 | torch 2.11 | CXX11 ABI True
-flash-attn @ https://github.com/lesj0610/flash-attention/releases/download/v2.8.3-cu12-torch2.11/flash_attn-2.8.3+cu12torch2.11cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
+#flash-attn @ https://github.com/lesj0610/flash-attention/releases/download/v2.8.3-cu12-torch2.11/flash_attn-2.8.3+cu12torch2.11cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
 
-# AIFS / anemoi stack
 anemoi-inference[huggingface]==0.8.3
 anemoi-models==0.9.3
 anemoi-utils==0.4.35.post3
@@ -17,7 +14,14 @@ earthkit-regrid==0.5.1
 ecmwf-opendata==0.3.29
 earthkit-data<1.0.0
 
-# Plotting
 gradio
 matplotlib
-cartopy
+cartopy
+
+# HF Space environment config:
+# Python: 3.12.12 (main, Feb 24 2026, 21:49:09) [GCC 14.2.0]
+#Platform: Linux-6.12.80-106.156.amzn2023.x86_64-x86_64-with-glibc2.41
+#Torch: 2.11.0+cu130
+#CUDA: 13.0
+#CXX11 ABI: True
+#GPU: NVIDIA RTX PRO 6000 Blackwell Server Edition