From 60ca8b08d0a0a3a9dd918c4f94806450d3e6898c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Mon, 24 Nov 2025 18:15:30 +0100
Subject: [PATCH] feat: add arty script to fix DiffRhythm transformers version
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Adds 'fix/diffrhythm-transformers' command to quickly downgrade
transformers library to 4.49.0 for DiffRhythm compatibility.

Usage: arty fix/diffrhythm-transformers

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 arty.yml | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/arty.yml b/arty.yml
index f5bf2e6..2156ee0 100644
--- a/arty.yml
+++ b/arty.yml
@@ -396,6 +396,38 @@ scripts:
     echo ""
     echo "Category: 🌸Pivoine/Audio"
 
+  fix/diffrhythm-transformers: |
+    echo "========================================="
+    echo "  Fix DiffRhythm Transformers Version"
+    echo "========================================="
+    echo ""
+    echo "Issue: Tensor dimension mismatch (32 vs 64) in rotary embeddings"
+    echo "Solution: Downgrade transformers to 4.49.0"
+    echo ""
+    echo "References:"
+    echo "  - https://github.com/billwuhao/ComfyUI_DiffRhythm/issues/44"
+    echo "  - https://github.com/billwuhao/ComfyUI_DiffRhythm/issues/48"
+    echo ""
+
+    cd /workspace/ComfyUI
+    source venv/bin/activate
+
+    echo "Current transformers version:"
+    pip show transformers | grep Version
+    echo ""
+
+    echo "Installing transformers==4.49.0..."
+    pip install transformers==4.49.0
+
+    echo ""
+    echo "✓ Transformers downgraded to 4.49.0"
+    echo ""
+    echo "Next steps:"
+    echo "  1. Restart ComfyUI: arty services/comfyui/restart"
+    echo "  2. Test DiffRhythm workflows"
+
+    deactivate
+
   setup/comfyui-extensions-deps: |
     echo "========================================="
     echo "  Installing ComfyUI Extensions Dependencies"