about vton_unet_train

[DevelMayCry-MrChen]

vton_unet_train is not exit

so

this code: is run error
unet_garm = UNetGarm2DConditionModel.from_pretrained(
UNET_PATH,
subfolder="unet_garm_train",
torch_dtype=torch.float32,
use_safetensors=True,
)

unet_vton = UNetVton2DConditionModel.from_pretrained(
UNET_PATH,
subfolder="unet_vton_train",
torch_dtype=torch.float32,
use_safetensors=True,
)

my code:
unet_garm = UNetGarm2DConditionModel.from_config(config='checkpoints/ootd/ootd_hd/checkpoint-36000/unet_garm/config.json').to(torch.float32)

unet_vton = UNetVton2DConditionModel.from_config(config='checkpoints/ootd/ootd_hd/checkpoint-36000/unet_vton/config.json').to(torch.float32)

is right?

now train loss:0.0285