deps for blip

This commit is contained in:
gbtb 2022-10-24 23:52:23 +10:00
parent 6f96b9d189
commit a1eefd2eda
3 changed files with 52 additions and 4 deletions

View File

@ -10,10 +10,6 @@
url = "github:CompVis/stable-diffusion?rev=69ae4b35e0a0f6ee1af8bb9a5d0016ccb27e36dc";
flake = false;
};
#codeformer-repo = {
# url = "github:sczhou/CodeFormer?rev=c5b4593074ba6214284d6acd5f1719b6c5d739af";
# flake = false;
#};
};
outputs = { self, nixpkgs, nixlib, stable-diffusion-repo }@inputs:
let

View File

@ -0,0 +1,26 @@
# WARNING: This file was automatically generated. You should avoid editing it.
# If you run pynixify again, the file will be either overwritten or
# deleted, and you will lose the changes you made to it.
{ buildPythonPackage, fetchPypi, lib, ninja, numpy, torch }:
buildPythonPackage rec {
pname = "fairscale";
version = "0.4.12";
src = fetchPypi {
inherit pname version;
sha256 = "0zjgdida0bxvi4dx1dgq0sfl6h7cjmkma5c1rxbwym97p8aqcs3g";
};
buildInputs = [ ninja ];
propagatedBuildInputs = [ torch numpy ];
# TODO FIXME
doCheck = false;
meta = with lib; {
description =
"FairScale: A PyTorch library for large-scale and high-performance training.";
};
}

26
packages/timm/default.nix Normal file
View File

@ -0,0 +1,26 @@
# WARNING: This file was automatically generated. You should avoid editing it.
# If you run pynixify again, the file will be either overwritten or
# deleted, and you will lose the changes you made to it.
{ buildPythonPackage, fetchPypi, huggingface-hub, lib, pyyaml, torch
, torchvision }:
buildPythonPackage rec {
pname = "timm";
version = "0.6.11";
src = fetchPypi {
inherit pname version;
sha256 = "199zgg574gijw52jn4fhfzaqlbwhq0z8kav4k34xifssnr18hmh9";
};
propagatedBuildInputs = [ torch torchvision pyyaml huggingface-hub ];
# TODO FIXME
doCheck = false;
meta = with lib; {
description = "(Unofficial) PyTorch Image Models";
homepage = "https://github.com/rwightman/pytorch-image-models";
};
}