Merge pull request #1 from timechess/main

Use poetry and Nix
CIntellifusion · Jun 6, 2024 · f74fe6c · f74fe6c
2 parents 604357b + b842af5
commit f74fe6c
Show file tree

Hide file tree

Showing 9 changed files with 2,677 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,2 @@
-/LatentDiffusion/data/__pycache__
+**/__pycache__
 *.pyc
diff --git a/LatentDiffusion/data/__pycache__/data_wrapper.cpython-38.pyc b/LatentDiffusion/data/__pycache__/data_wrapper.cpython-38.pyc
diff --git a/LatentDiffusion/models/__pycache__/unet.cpython-38.pyc b/LatentDiffusion/models/__pycache__/unet.cpython-38.pyc
diff --git a/LatentDiffusion/schedulers/__pycache__/ddpm.cpython-38.pyc b/LatentDiffusion/schedulers/__pycache__/ddpm.cpython-38.pyc
diff --git a/flake.lock b/flake.lock
diff --git a/flake.nix b/flake.nix
@@ -0,0 +1,11 @@
+{
+  inputs = {
+    flake-utils.url = "github:numtide/flake-utils";
+    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable-small";
+  };
+
+  outputs = { self, nixpkgs, flake-utils }:
+    flake-utils.lib.eachDefaultSystem (system:
+      let pkgs = nixpkgs.legacyPackages.${system};
+      in { devShells.default = import ./shell.nix { inherit pkgs system; }; });
+}
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -0,0 +1,35 @@
+[tool.poetry]
+name = "simplediffusion"
+version = "0.1.0"
+description = ""
+#authors = ["Your Name <[email protected]>"]
+license = "MIT"
+readme = "README.md"
+package-mode = false
+
+[tool.poetry.dependencies]
+python = ">=3.9,<3.13"
+pytorch-lightning = "^1.8.3"
+torchmetrics = "^0.11.4"
+datasets = "^2.18.0"
+transformers = "4.30.2"
+diffusers = "^0.20.0"
+torch = { version = "2.0.1", source="torch"}
+torchvision = { version = "0.15.2", source="torch"}
+opencv-python = "^4.9.0.80"
+tensorboard = "^2.16.2"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[[tool.poetry.source]]
+name = "torch"
+url = "https://download.pytorch.org/whl/cu118/"
+priority = "supplemental"
+
+
+[[tool.poetry.source]]
+name = "mirrors"
+url = "https://pypi.tuna.tsinghua.edu.cn/simple/"
+priority = "primary"
diff --git a/shell.nix b/shell.nix
@@ -0,0 +1,14 @@
+{ pkgs, ... }:
+
+(pkgs.mkShell rec {
+
+  buildInputs = (with pkgs; [ python39 poetry gcc gnumake zsh zlib libGL glib]);
+
+  LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath [ pkgs.stdenv.cc.cc];
+
+  shellHook = ''
+  export LD_LIBRARY_PATH="${pkgs.lib.makeLibraryPath buildInputs}:$LD_LIBRARY_PATH"
+  poetry env use ${pkgs.python39}/bin/python
+  poetry install
+  '';
+})