Update

app.py

@@ -12,8 +12,8 @@ from transformers import KyutaiSpeechToTextForConditionalGeneration, KyutaiSpeec
 DESCRIPTION = "# Kyutai STT 2.6B EN"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "kyutai/stt-2.6b-en"
-model = KyutaiSpeechToTextForConditionalGeneration.from_pretrained(model_id, device_map=device)
+model_id = "kyutai/stt-2.6b-en-trfs"
+model = KyutaiSpeechToTextForConditionalGeneration.from_pretrained(model_id, device_map=device, torch_dtype="auto")
 processor = KyutaiSpeechToTextProcessor.from_pretrained(model_id)
 
 SAMPLE_RATE = 24000

pyproject.toml

@@ -11,7 +11,7 @@ dependencies = [
     "librosa>=0.11.0",
     "spaces>=0.37.1",
     "torch==2.5.1",
-    "transformers",
+    "transformers>=4.53.0",
 ]
 
 [tool.ruff]
@@ -54,6 +54,3 @@ convention = "google"
 
 [tool.ruff.format]
 docstring-code-format = true
-
-[tool.uv.sources]
-transformers = { git = "https://github.com/huggingface/transformers", rev = "v4.52.4-Kyutai-STT-preview" }

requirements.txt

@@ -280,7 +280,7 @@ tqdm==4.67.1
     # via
     #   huggingface-hub
     #   transformers
-transformers @ git+https://github.com/huggingface/transformers@6bdd4ec95264e5d8f219cfe4ee29ea9b42474bb7
+transformers==4.53.0
     # via kyutai-stt-2-6b-en (pyproject.toml)
 triton==3.1.0
     # via torch

uv.lock

@@ -554,7 +554,7 @@ requires-dist = [
     { name = "librosa", specifier = ">=0.11.0" },
     { name = "spaces", specifier = ">=0.37.1" },
     { name = "torch", specifier = "==2.5.1" },
-    { name = "transformers", git = "https://github.com/huggingface/transformers?rev=v4.52.4-Kyutai-STT-preview" },
+    { name = "transformers", specifier = ">=4.53.0" },
 ]
 
 [[package]]
@@ -2058,8 +2058,8 @@ wheels = [
 
 [[package]]
 name = "transformers"
-version = "4.53.0.dev0"
-source = { git = "https://github.com/huggingface/transformers?rev=v4.52.4-Kyutai-STT-preview#6bdd4ec95264e5d8f219cfe4ee29ea9b42474bb7" }
+version = "4.53.0"
+source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "filelock" },
     { name = "huggingface-hub" },
@@ -2072,6 +2072,10 @@ dependencies = [
     { name = "tokenizers" },
     { name = "tqdm" },
 ]
+sdist = { url = "https://files.pythonhosted.org/packages/e8/40/f2d2c3bcf5c6135027cab0fd7db52f6149a1c23acc4e45f914c43d362386/transformers-4.53.0.tar.gz", hash = "sha256:f89520011b4a73066fdc7aabfa158317c3934a22e3cd652d7ffbc512c4063841", size = 9177265, upload-time = "2025-06-26T16:10:54.729Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/5e/0c/68d03a38f6ab2ba2b2829eb11b334610dd236e7926787f7656001b68e1f2/transformers-4.53.0-py3-none-any.whl", hash = "sha256:7d8039ff032c01a2d7f8a8fe0066620367003275f023815a966e62203f9f5dd7", size = 10821970, upload-time = "2025-06-26T16:10:51.505Z" },
+]
 
 [[package]]
 name = "triton"