pawlo2013 commited on
Commit
3a0cf74
·
1 Parent(s): 0b2b0ab

adde requirements and safetensor weights

Browse files
Files changed (3) hide show
  1. .gitignore +1 -1
  2. models/model-epoch_80.st +3 -0
  3. requirements.txt +14 -0
.gitignore CHANGED
@@ -17,7 +17,7 @@ __pycache__/
17
 
18
  *.bin
19
 
20
- *.st
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