Spaces:
Running
Running
adde requirements and safetensor weights
Browse files- .gitignore +1 -1
- models/model-epoch_80.st +3 -0
- requirements.txt +14 -0
.gitignore
CHANGED
@@ -17,7 +17,7 @@ __pycache__/
|
|
17 |
|
18 |
*.bin
|
19 |
|
20 |
-
|
21 |
|
22 |
.models/model-epoch_80.st
|
23 |
.history/
|
|
|
17 |
|
18 |
*.bin
|
19 |
|
20 |
+
|
21 |
|
22 |
.models/model-epoch_80.st
|
23 |
.history/
|
models/model-epoch_80.st
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5ef669b8e863309008a02441de07cc2f09d25ebfa03246fc0f13dfa9fe9ec91d
|
3 |
+
size 607817532
|
requirements.txt
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
einops
|
2 |
+
datasets
|
3 |
+
matplotlib
|
4 |
+
tqdm
|
5 |
+
accelerate
|
6 |
+
jax[cpu]
|
7 |
+
torchinfo
|
8 |
+
wandb
|
9 |
+
ema_pytorch
|
10 |
+
lpips
|
11 |
+
pyyaml
|
12 |
+
diffusers
|
13 |
+
transformers
|
14 |
+
torch-ema
|