fix: ai notebook sentencepipe incompatibility with apple devices
This commit is contained in:
parent
e27097fb34
commit
26c8423a47
4 changed files with 160 additions and 38 deletions
|
|
@ -22,12 +22,11 @@ let
|
|||
scikit-image
|
||||
scikit-learn
|
||||
seaborn
|
||||
sentencepiece
|
||||
tokenizers
|
||||
transformers
|
||||
];
|
||||
linuxOnlyPackages = lib.optionals pkgs.stdenv.isLinux
|
||||
(with ps; [ torch torchaudio torchvision ]);
|
||||
(with ps; [ sentencepiece torch torchaudio torchvision ]);
|
||||
in basePackages ++ linuxOnlyPackages);
|
||||
in pkgs.mkShell {
|
||||
name = "ai-notebook";
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue