···

       
24
       
24
       
 
       
  pandas,

     

       
25
       
25
       
 
       
  transformers,

     

       
26
       
26
       
 
       
  webdataset,

     

       
27
       
27
       
-
       


     

       
28
       
28
       
-
       
  stdenv,

     

       
29
       
27
       
 
       
}:

     

       
30
       
28
       
 
       
buildPythonPackage rec {

     

       
31
       
29
       
 
       
  pname = "open-clip-torch";

     

       
32
       
32
       
-
       
  version = "3.1.0";

     

       
30
       
30
       
+
       
  version = "3.2.0";

     

       
33
       
31
       
 
       
  pyproject = true;

     

       
34
       
32
       
 
       


     

       
35
       
33
       
 
       
  src = fetchFromGitHub {

     

       
36
       
34
       
 
       
    owner = "mlfoundations";

     

       
37
       
35
       
 
       
    repo = "open_clip";

     

       
38
       
36
       
 
       
    tag = "v${version}";

     

       
39
       
39
       
-
       
    hash = "sha256-xDXxhncX0l9qwbV4Gk4rOROH6Qyit/FSTgjrg4Vbk1s=";

     

       
37
       
37
       
+
       
    hash = "sha256-k4/u0XtfBmPSVKfEK3wHqJXtKAuUNkUnk1TLG2S6PPs=";

     

       
40
       
38
       
 
       
  };

     

       
41
       
39
       
 
       


     

       
42
       
40
       
 
       
  build-system = [ pdm-backend ];

     
···

       
77
       
75
       
 
       
    "test_inference_with_data"

     

       
78
       
76
       
 
       
    "test_pretrained_text_encoder"

     

       
79
       
77
       
 
       
    "test_training_mt5"

     

       
78
       
78
       
+
       


     

       
80
       
79
       
 
       
    # fails due to type errors

     

       
81
       
80
       
 
       
    "test_num_shards"

     

       
82
       
82
       
-
       
  ]

     

       
83
       
83
       
-
       
  ++ lib.optionals (stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isLinux) [

     

       
81
       
81
       
+
       


     

       
82
       
82
       
+
       
    # hangs forever

     

       
84
       
83
       
 
       
    "test_training"

     

       
85
       
85
       
-
       
    "test_training_coca"

     

       
86
       
86
       
-
       
    "test_training_unfreezing_vit"

     

       
87
       
87
       
-
       
    "test_training_clip_with_jit"

     

       
88
       
84
       
 
       
  ];

     

       
89
       
85
       
 
       


     

       
90
       
86
       
 
       
  meta = {