Update ootd/inference_ootd.py

ootd/inference_ootd.py

@@ -43,13 +43,13 @@ class OOTDiffusion:
 
         unet_garm = UNetGarm2DConditionModel.from_pretrained(
             UNET_PATH,
-            subfolder="ootd_hd/checkpoint-36000/unet_garm",
+            subfolder="unet_garm",
             torch_dtype=torch.float16,
             use_safetensors=True,
         )
         unet_vton = UNetVton2DConditionModel.from_pretrained(
             UNET_PATH,
-            subfolder="ootd_hd/checkpoint-36000/unet_vton",
+            subfolder="unet_vton",
             torch_dtype=torch.float16,
             use_safetensors=True,
         )