commit 4ebe160b24c0891b3a0d462b3df8cd4866f6ab10 · pyrox.dev/nixpkgs

+10 -1
pkgs/development/python-modules/mlx-lm/default.nix
···

       2
       2
        
         lib,

     

       3
       3
        
         buildPythonPackage,

     

       4
       4
        
         fetchFromGitHub,

     

       5
       5
       +
       

     

       6
       6
       +
         # build-system

     

       5
       7
        
         setuptools,

     

       8
       8
       +
       

     

       9
       9
       +
         # dependencies

     

       6
       10
        
         jinja2,

     

       7
       11
        
         mlx,

     

       8
       12
        
         numpy,

     

       9
       13
        
         protobuf,

     

       10
       14
        
         pyyaml,

     

       11
       15
        
         transformers,

     

       16
       16
       +
       

     

       17
       17
       +
         # tests

     

       18
       18
       +
         lm-eval,

     

       12
       19
        
         sentencepiece,

     

       13
       20
        
         pytestCheckHook,

     

       14
       21
        
         writableTmpDirAsHomeHook,

     
···

       40
       47
        
         ];

     

       41
       48
        
       

     

       42
       49
        
         nativeCheckInputs = [

     

       43
       43
       -
           writableTmpDirAsHomeHook

     

       50
       50
       +
           lm-eval

     

       44
       51
        
           pytestCheckHook

     

       45
       52
        
           sentencepiece

     

       53
       53
       +
           writableTmpDirAsHomeHook

     

       46
       54
        
         ];

     

       47
       55
        
       

     

       48
       56
        
         pythonImportsCheck = [

     
···

       62
       70
        
           "tests/test_prompt_cache.py::TestPromptCache::test_cache_with_generate"

     

       63
       71
        
           "tests/test_prompt_cache.py::TestPromptCache::test_trim_cache_with_generate"

     

       64
       72
        
           # RuntimeError: [metal_kernel] No GPU back-end.

     

       73
       73
       +
           "tests/test_losses.py"

     

       65
       74
        
           "tests/test_models.py::TestModels::test_bitnet"

     

       66
       75
        
         ];

     

       67
       76