BeveledCube commited on
Commit
1cfa651
·
1 Parent(s): 3b3a783
Files changed (2) hide show
  1. pyproject.toml +23 -0
  2. start.sh +5 -0
pyproject.toml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [tool.poetry]
2
+ name = "img-gen"
3
+ version = "0.0.1"
4
+ description = ""
5
+ authors = ["CubeBeveled <[email protected]>"]
6
+ readme = "README.md"
7
+
8
+ [tool.poetry.dependencies]
9
+ python = "^3.11"
10
+ torch = "^2.2.1"
11
+ numpy = "^1.26.4"
12
+ einops = "^0.7.0"
13
+ torchvision = "^0.17.1"
14
+ tqdm = "^4.66.2"
15
+ diffusers = "^0.27.2"
16
+ accelerate = "^0.28.0"
17
+ transformers = "^4.39.1"
18
+ pillow = "^10.2.0"
19
+
20
+
21
+ [build-system]
22
+ requires = ["poetry-core"]
23
+ build-backend = "poetry.core.masonry.api"
start.sh ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ pip install --upgrade pip
2
+ pip install -r requirements.txt
3
+ poetry install --no-root
4
+
5
+ python main.py