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'
|