spaces 0.50.dev0__tar.gz → 0.50.dev2__tar.gz

{spaces-0.50.dev0 → spaces-0.50.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spaces
-Version: 0.50.dev0
+Version: 0.50.dev2
 Summary: Utilities for Hugging Face Spaces
 Author: Charles Bensimon
 Author-email: Charles Bensimon <charles@huggingface.co>

{spaces-0.50.dev0 → spaces-0.50.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spaces"
-version = "0.50.dev0"
+version = "0.50.dev2"
 description = "Utilities for Hugging Face Spaces"
 readme = "README.md"
 requires-python = ">=3.10,<3.15"

{spaces-0.50.dev0 → spaces-0.50.dev2}/spaces/__init__.py

@@ -8,7 +8,7 @@ from typing import TYPE_CHECKING
 from typing import Callable
 
 
-# Prevent gradio from importing spaces
+# Prevent partially-imported gradio from importing spaces
 if (gr := sys.modules.get('gradio')) is not None: # pragma: no cover
     try:
         gr.Blocks

{spaces-0.50.dev0 → spaces-0.50.dev2}/spaces/zero/__init__.py

@@ -6,6 +6,10 @@ from importlib import metadata
 from packaging.version import Version
 
 from ..config import Config
+from .context import ZeroGPUError
+from .context import ZeroGPUErrorResponse
+from .context import ZeroGPUException
+from .context import ZeroGPULog
 from .middleware import ZeroGPUMiddleware
 from .lifecycle import zerogpu_mode_enable
 from .lifecycle import zerogpu_mode_enabled
@@ -18,11 +22,23 @@ def _maybe_enable_gradio_compat():
     except metadata.PackageNotFoundError: # pragma: no cover
         pass
     else:
-        if Version(gradio_version) < Version('6.13'):
+        if Version(gradio_version) < Version('6.14'):
             from . import gradio
-            gradio.patch(zerogpu_middleware_class=ZeroGPUMiddleware)
+            gradio.patch()
 
 
 if Config.zero_gpu:
     zerogpu_mode_enable()
     _maybe_enable_gradio_compat()
+
+
+__all__ = [
+    'ZeroGPUError',
+    'ZeroGPUErrorResponse',
+    'ZeroGPUException',
+    'ZeroGPULog',
+    'ZeroGPUMiddleware',
+    'zerogpu_mode_enable',
+    'zerogpu_mode_enabled',
+    'zerogpu_startup',
+]

{spaces-0.50.dev0 → spaces-0.50.dev2}/spaces/zero/gradio.py

@@ -37,7 +37,7 @@ class WorkerContextualizer:
         LocalContext.progress.set(self.progress)
 
 
-def patch(zerogpu_middleware_class: Type[ZeroGPUMiddleware]):
+def patch():
 
     try:
         Success = gr.Success # pyright: ignore[reportAttributeAccessIssue] (Gradio<5.10)
@@ -70,7 +70,7 @@ def patch(zerogpu_middleware_class: Type[ZeroGPUMiddleware]):
     def create_app(*args, **kwargs):
         app = orig_create_app(*args, **kwargs)
         app.add_middleware(
-            zerogpu_middleware_class,
+            ZeroGPUMiddleware,
             exception_mapper=lambda err, exc: (
                 setattr(exc, 'print_exception', False) or exc
                 if isinstance(exc, gr.Error)

{spaces-0.50.dev0 → spaces-0.50.dev2}/spaces/zero/lifecycle.py

@@ -1,6 +1,12 @@
 """
 """
 
+import shutil
+from pathlib import Path
+
+from ..config import Config
+
+
 BAD_FORK_EXCEPTION_MESSAGE = (
     "CUDA has been initialized before importing the `spaces` package. "
     "Try importing `spaces` before any other CUDA-related package."
@@ -16,6 +22,8 @@ def zerogpu_mode_enable():
     if torch.is_in_bad_fork():
         raise RuntimeError(BAD_FORK_EXCEPTION_MESSAGE)
     torch.patch()
+    shutil.rmtree(Config.zerogpu_offload_dir, ignore_errors=True)
+    Path(Config.zerogpu_offload_dir).mkdir(parents=True)
     _zerogpu_mode_enabled = True
 
 

{spaces-0.50.dev0 → spaces-0.50.dev2}/spaces/zero/middleware.py

@@ -1,7 +1,7 @@
 """
 """
-from abc import ABC
-from abc import abstractmethod
+
+import asyncio.threads
 from typing import Any
 from typing import Awaitable
 from typing import Callable
@@ -61,7 +61,7 @@ class ZeroGPUMiddleware:
             async def wrapped_receive() -> ASGIReceiveEvent:
                 message = await asgi_receive()
                 if message['type'] == 'lifespan.startup':
-                    zerogpu_startup()
+                    await asyncio.threads.to_thread(zerogpu_startup)
                 return message
             await asgi_app(asgi_scope, wrapped_receive, asgi_send)
             return # pragma: no cover

{spaces-0.50.dev0 → spaces-0.50.dev2}/spaces/zero/torch/patching.py

@@ -5,13 +5,11 @@
 import gc
 import multiprocessing
 import os
-import shutil
 from collections import defaultdict
 from concurrent.futures import ProcessPoolExecutor
 from concurrent.futures import ThreadPoolExecutor
 from contextlib import nullcontext
 from contextvars import copy_context
-from pathlib import Path
 from typing import Any
 from typing import Callable
 
@@ -391,8 +389,6 @@ def _pack(offload_dir: str):
 def pack():
     if len(cuda_aliases) == 0:
         return 0
-    shutil.rmtree(Config.zerogpu_offload_dir, ignore_errors=True)
-    Path(Config.zerogpu_offload_dir).mkdir(parents=True)
     mmap_addrs = [tensor.data_ptr() for tensor in cuda_aliases.values() if tensor is not None]
     with unmap_capture(mmap_addrs) as unmapped_paths:
         total_size = _pack(Config.zerogpu_offload_dir)