Initial public ModelForge release

2026-09-01 21:30:16 +02:00
commit 7082ab955a
490 changed files with 104252 additions and 0 deletions
@@ -0,0 +1,54 @@
+"""Use 64-bit scheduler and worker generations.
+
+Revision ID: 20260826_0015
+Revises: 20260826_0014
+"""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+
+from alembic import op
+
+revision: str = "20260826_0015"
+down_revision: str | None = "20260826_0014"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    op.alter_column(
+        "residency_allocations",
+        "generation",
+        existing_type=sa.Integer(),
+        type_=sa.BigInteger(),
+        existing_nullable=False,
+        existing_server_default="1",
+    )
+    op.alter_column(
+        "serving_gpu_leases",
+        "generation",
+        existing_type=sa.Integer(),
+        type_=sa.BigInteger(),
+        existing_nullable=False,
+        existing_server_default="1",
+    )
+
+
+def downgrade() -> None:
+    op.alter_column(
+        "serving_gpu_leases",
+        "generation",
+        existing_type=sa.BigInteger(),
+        type_=sa.Integer(),
+        existing_nullable=False,
+        existing_server_default="1",
+    )
+    op.alter_column(
+        "residency_allocations",
+        "generation",
+        existing_type=sa.BigInteger(),
+        type_=sa.Integer(),
+        existing_nullable=False,
+        existing_server_default="1",
+    )