Commit
Β·
673064b
1
Parent(s):
26c80cc
fix: Update attention checkpoint path in app.py for CatVTON pipeline
Browse files- Changed the attention checkpoint path from the previous repository to the MaskFree version for improved model performance.
- Updated the attention checkpoint version to a more general reference, enhancing compatibility with future updates.
app.py
CHANGED
@@ -151,8 +151,8 @@ catvton_mf_repo = "zhengchong/CatVTON-MaskFree"
|
|
151 |
repo_path_mf = snapshot_download(repo_id=catvton_mf_repo)
|
152 |
pipeline_p2p = CatVTONPix2PixPipeline(
|
153 |
base_ckpt=args.p2p_base_model_path,
|
154 |
-
attn_ckpt=
|
155 |
-
attn_ckpt_version="mix
|
156 |
weight_dtype=init_weight_dtype(args.mixed_precision),
|
157 |
use_tf32=args.allow_tf32,
|
158 |
device='cuda'
|
|
|
151 |
repo_path_mf = snapshot_download(repo_id=catvton_mf_repo)
|
152 |
pipeline_p2p = CatVTONPix2PixPipeline(
|
153 |
base_ckpt=args.p2p_base_model_path,
|
154 |
+
attn_ckpt=repo_path_mf,
|
155 |
+
attn_ckpt_version="mix",
|
156 |
weight_dtype=init_weight_dtype(args.mixed_precision),
|
157 |
use_tf32=args.allow_tf32,
|
158 |
device='cuda'
|