overrides: add open-clip-torch

nix-community · Oct 2, 2024 · 1cc00ec · 1cc00ec
1 parent 62bde23
commit 1cc00ec
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/overrides/default.nix b/overrides/default.nix
@@ -1982,6 +1982,30 @@ lib.composeManyExtensions [
         }
       );
 
+      open-clip-torch = prev.open-clip-torch.overridePythonAttrs (
+        # The sdist from pypi doesn't contain the requirements.txt
+        old:
+        lib.optionalAttrs (!(old.src.isWheel or false)) (
+          let
+            githubHash =
+              {
+                "2.20.0" = "sha256-Ca4oi2LqleIFAGBJB7YIi4nXe2XhOP6ErDFXgXtJLxM=";
+              }
+              .${old.version} or lib.fakeHash;
+
+            src = pkgs.fetchFromGitHub {
+              owner = "mlfoundations";
+              repo = "open_clip";
+              rev = "v${old.version}";
+              sha256 = githubHash;
+            };
+          in
+          {
+            inherit src;
+          }
+        )
+      );
+
       orjson = prev.orjson.overridePythonAttrs (
         old: lib.optionalAttrs (!(old.src.isWheel or false)) (
           let