From 5486aa2fd721cd3a9c9cb9b1f9cc167bf8bfc0a3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= <me@danieldk.eu>
Date: Thu, 21 Nov 2024 10:44:24 +0000
Subject: [PATCH] python3Packages.outlines: 0.1.4 -> 0.1.3

Temporary downgrade to avoid
https://github.com/dottxt-ai/outlines-core/issues/95
---
 flake.nix   |  1 +
 overlay.nix | 30 ++++++++++++++++++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/flake.nix b/flake.nix
index 860b4db..4dde3c2 100644
--- a/flake.nix
+++ b/flake.nix
@@ -64,6 +64,7 @@
               marlin-kernels
               moe-kernels
               opentelemetry-instrumentation-grpc
+              outlines
               punica-kernels
               torch
               ;
diff --git a/overlay.nix b/overlay.nix
index 1a6f319..8dfb64b 100644
--- a/overlay.nix
+++ b/overlay.nix
@@ -54,6 +54,36 @@ final: prev: {
           }
         );
 
+        # Temporarily downgrade outlines to work around https://github.com/dottxt-ai/outlines-core/issues/95
+
+        outlines = python-super.outlines.overrideAttrs (
+          _: prevAttrs: rec {
+            version = "0.1.3";
+            name = "${prevAttrs.pname}-${version}";
+
+            src = prev.fetchFromGitHub {
+              owner = "dottxt-ai";
+              repo = prevAttrs.pname;
+              rev = version;
+              hash = "sha256-OACSwdkh46TwTtgsy4HNZbx62UPJJ/Lq1JJ0fFPr9mQ=";
+            };
+          }
+        );
+
+        outlines-core = python-super.outlines-core.overrideAttrs (
+          _: prevAttrs: rec {
+            version = "0.1.14";
+            name = "${prevAttrs.pname}-${version}";
+
+            src = prev.fetchFromGitHub {
+              owner = "dottxt-ai";
+              repo = prevAttrs.pname;
+              rev = version;
+              hash = "sha256-1S1KCTmHRc/5vviRd2fFFh/Sx1OKWDFlrQusMFBjLck=";
+            };
+          }
+        );
+
         mamba-ssm = callPackage ./pkgs/python-modules/mamba-ssm { };
 
         punica-kernels = callPackage ./pkgs/python-modules/punica-kernels { };