OOTDiffusion icon indicating copy to clipboard operation
OOTDiffusion copied to clipboard

where can i download the ootd_dc model?

Open frankchieng opened this issue 1 year ago • 2 comments

it seems like only the ootd_hd upper body model has been released,when and how can i get the full body ootd_dc model?i appreciate for your hard work

frankchieng avatar Feb 21 '24 11:02 frankchieng

Same, I duplicated ootd_hd in /checkpoints/ootd/ and rename to ootd_dc

Like this: image

But lower body and dress will be defective :))) any fix it?

pvbang avatar Feb 22 '24 01:02 pvbang

I met the same issue? where we can download the "OOTDiffusion/checkpoints/ootd/ootd_dc" ?

Nick1211 avatar Feb 26 '24 07:02 Nick1211

The ootd_dc model is available now. Thanks for your patience : )

levihsu avatar Mar 04 '24 07:03 levihsu

The ootd_dc model is available now. Thanks for your patience : )

when i run the gradio_ootd.py, it raise error as below: raise ValueError( ValueError: Cannot load ../checkpoints/ootd/ootd_dc/checkpoint-36000 because conv_in.weight expected shape tensor(..., device='meta', size=(320, 4, 3, 3)), but got torch.Size([320, 8, 3, 3]). If you want to instead overwrite randomly initialized weights, please make sure to pass both low_cpu_mem_usage=False and ignore_mismatched_sizes=True. For more information, see also: https://github.com/huggingface/diffusers/issues/1619#issuecomment-1345604389 as an example.

i tried to modify the ootd/inference_ootd_dc.py with

    unet_garm = UNetGarm2DConditionModel.from_pretrained(
          UNET_PATH,
          subfolder="unet_garm",
          torch_dtype=torch.float16,
          use_safetensors=True,
          low_cpu_mem_usage=False,
          ignore_mismatched_sizes=True
      )
      unet_vton = UNetVton2DConditionModel.from_pretrained(
          UNET_PATH,
          subfolder="unet_vton",
          torch_dtype=torch.float16,
          use_safetensors=True,
          low_cpu_mem_usage=False,
          ignore_mismatched_sizes=True
      )

then i can run gradio successfully, but the Full-body generation with DC model renders quality is so badly, any clue how i can solve this issue?

frankchieng avatar Mar 05 '24 09:03 frankchieng