update

app.py

@@ -39,7 +39,7 @@ DTYPE = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
 path = "finegrain/merged-turbo-light-switcher"
 model = light_switcher_lite.Model.from_pretrained(path, device=DEVICE_CPU, dtype=DTYPE)
 
-model.to(DEVICE)
+model = model.to(DEVICE)
 
 
 @spaces.GPU

pyproject.toml

@@ -8,13 +8,15 @@ authors = [
 dependencies = [
     "gradio>=5.27.1",
     "pillow>=11.3.0",
-    "refiners @ git+https://github.com/finegrain-ai/refiners",
     "spaces>=0.29.3",
     "numpy<2.0.0",
     "pi-heif>=1.1.0",
     "diffusers>=0.35.1",
     "accelerate>=1.10.1",
+    "finegrain_toolbox @ git+https://github.com/finegrain-ai/finegrain-toolbox",
 ]
+# ZeroGPU is stuck with 3.10
+requires-python = ">= 3.10"
 
 [tool.ruff]
 line-length = 120

requirements.txt

@@ -1,6 +1,6 @@
 spaces>=0.35.0
-git+https://github.com/huggingface/diffusers@97abdd2210a540c2e71aee63c80a22723031cd57
-git+https://github.com/finegrain-ai/refiners@cfe8b66ba4f8a906583850ac25e9e89cb83a44b9
+diffusers>=0.35.0
+git+https://github.com/finegrain-ai/finegrain-toolbox@906e408c3c0e4e30b307fe9290b10d73e5b81c97
 numpy<2.0.0
 pillow>=11.3.0
 pi_heif>=1.1.0