ootd

f6008757 · mashun1 · 151e8d3e · f6008757 · f6008757
Commit f6008757 authored May 22, 2024 by mashun1
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

README.md README.md +1 -1

ootd/inference_ootd_hd.py ootd/inference_ootd_hd.py +2 -2

No files found.
--- a/README.md
+++ b/README.md
@@ -151,7 +151,7 @@ https://hf-mirror.com/openai/clip-vit-large-patch14/tree/main
 ### webui
    cd OOTDiffusion/run
-    python gradio_output.py
+    python gradio_ootd.py
 ## result

--- a/ootd/inference_ootd_hd.py
+++ b/ootd/inference_ootd_hd.py
@@ -30,8 +30,8 @@ sys.path.append(str(OOTD_ROOT))
 # VIT_PATH = "../checkpoints/clip-vit-large-patch14"
 VIT_PATH = os.path.join(OOTD_ROOT, "checkpoints/clip-vit-large-patch14")
 VAE_PATH = "../checkpoints/ootd"
-UNET_PATH = "../checkpoints/ootd/ootd_hd/checkpoint-36000"
+# UNET_PATH = "../checkpoints/ootd/ootd_hd/checkpoint-36000"
-# UNET_PATH = "../train/checkpoints"
+UNET_PATH = "../train/checkpoints"
 MODEL_PATH = "../checkpoints/ootd"
 class OOTDiffusionHD: