Hugging Face's logo

next-social
/
ComfyUI_Seg_VITON2

Model card Files Community
ifwi commited on
Commit
aca659c
·
1 Parent(s): d4ed190

add more

Browse files
Files changed (7) hide show
  1. .gitattributes +34 -0
  2. .gitignore +3 -0
  3. checkpoints/VITONHD.ckpt +3 -0
  4. checkpoints/cloth_segm_u2net_latest.pth +3 -0
  5. checkpoints/segformer_b2_clothes +1 -0
  6. ldm/models/diffusion/ddpm.py +2 -2
  7. requirements.txt +11 -9
.gitattributes ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
.gitignore ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ .idea/
2
+ **/__pycache__
3
+ .cog/
checkpoints/VITONHD.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c98bfe02e04c9cc7bc37c5dc583f3c5a442fbd65d3555cb7ba6ad0154b51ca7
3
+ size 7355653199
checkpoints/cloth_segm_u2net_latest.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f71fad2bc11789a996acc507d1a5a1602ae0edefc2b9aba1cd198be5cc9c1a44
3
+ size 176625341
checkpoints/segformer_b2_clothes ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit f6ac72992f938a1d0073fb5e5a06fd781f19f9a2
ldm/models/diffusion/ddpm.py CHANGED
@@ -23,8 +23,8 @@ from torchvision.transforms.functional import resize
23
  import torchvision.transforms as T
24
  import random
25
  import torch.nn.functional as F
26
- from diffusers.models.autoencoder_kl import AutoencoderKLOutput
27
- from diffusers.models.vae import DecoderOutput
28
 
29
  from ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
30
  from ldm.modules.ema import LitEma
 
23
  import torchvision.transforms as T
24
  import random
25
  import torch.nn.functional as F
26
+ from diffusers.models.autoencoders.autoencoder_kl import AutoencoderKLOutput
27
+ from diffusers.models.autoencoders.vae import DecoderOutput
28
 
29
  from ldm.util import log_txt_as_img, exists, default, ismap, isimage, mean_flat, count_params, instantiate_from_config
30
  from ldm.modules.ema import LitEma
requirements.txt CHANGED
@@ -1,12 +1,14 @@
1
- pytorch-lightning==1.5.0
 
2
  einops
3
- opencv-python==4.7.0.72
4
  matplotlib
5
  omegaconf
6
- transformers==4.33.2
7
- xformers==0.0.19
8
- triton==2.0.0
9
- open-clip-torch==2.19.0
10
- diffusers==0.20.2
11
- scipy==1.10.1
12
- rembg
 
 
1
+ pytorch_lightning==v1.7.7
2
+ torchmetrics==0.11.4
3
  einops
4
+ opencv-python-headless
5
  matplotlib
6
  omegaconf
7
+ transformers
8
+ xformers
9
+ triton
10
+ open-clip-torch
11
+ diffusers
12
+ scipy
13
+ rembg
14
+ scikit-image