This commit is contained in:
gbtb 2023-02-28 22:39:23 +10:00
parent ba66c60f1e
commit 8d9bc7ba75
5 changed files with 8 additions and 7 deletions

View File

@ -52,7 +52,7 @@
datasets
click
pypatchmatch
#torchsde
torchsde
trampoline
send2trash
flask

View File

@ -4,7 +4,7 @@
{ accelerate, buildPythonPackage, clip-anytorch, einops, fetchPypi, jsonmerge
, kornia, lib, pillow, resize-right, scikit-image, scipy, torch, torchdiffeq
, torchvision, tqdm, wandb, clean-fid }:
, torchvision, tqdm, wandb, clean-fid, torchsde }:
buildPythonPackage rec {
pname = "k-diffusion";
@ -28,6 +28,7 @@ buildPythonPackage rec {
torch
torchdiffeq
torchvision
torchsde
tqdm
wandb
clean-fid

View File

@ -6,11 +6,11 @@
buildPythonPackage rec {
pname = "kornia";
version = "0.6.8";
version = "0.6.10";
src = fetchPypi {
inherit pname version;
sha256 = "0mlcnpqfd411hzn592lgz8pk3asdanks68ld1q1lzay0acjf1189";
sha256 = "0mlcapqfd411hzn592lgz8pk3asdanks68ld1q1lzay0acjf1189";
};
buildInputs = [ pytest-runner ];

View File

@ -2,7 +2,7 @@
# If you run pynixify again, the file will be either overwritten or
# deleted, and you will lose the changes you made to it.
{ boltons, buildPythonPackage, fetchPypi, lib, numpy, scipy, torch }:
{ boltons, buildPythonPackage, fetchPypi, lib, numpy, scipy, torch, trampoline }:
buildPythonPackage rec {
pname = "torchsde";
@ -13,7 +13,7 @@ buildPythonPackage rec {
sha256 = "0swhs2rp1hqi7qm55xh2rbcrz3qp8b2a1kvib95pmlqhavhyjar2";
};
propagatedBuildInputs = [ boltons numpy scipy torch ];
propagatedBuildInputs = [ boltons numpy scipy torch trampoline ];
# TODO FIXME
doCheck = false;

View File

@ -13,7 +13,7 @@ buildPythonPackage rec {
owner = "ferreum";
repo = pname;
rev = "6ff003ed89abc4b64587227d10a6a8ba48309a83";
sha256 = "";
sha256 = "sha256-1GK0MOF1uHhbT8qUQzr32B5HY8x3Nc0SvkjP0C21V6k=";
};
propagatedBuildInputs = [ ];