Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
6025240
1
Parent(s):
bde7aa3
fix: Update attention checkpoint version in CatVTON pipeline
Browse files- Changed the attention checkpoint version from "mix" to "mix-48k-1024" in app.py to improve model performance and ensure compatibility with the latest updates.
app.py
CHANGED
@@ -153,7 +153,7 @@ repo_path_mf = snapshot_download(repo_id=catvton_mf_repo, use_auth_token=access_
|
|
153 |
pipeline_p2p = CatVTONPix2PixPipeline(
|
154 |
base_ckpt=args.p2p_base_model_path,
|
155 |
attn_ckpt=repo_path_mf,
|
156 |
-
attn_ckpt_version="mix",
|
157 |
weight_dtype=init_weight_dtype(args.mixed_precision),
|
158 |
use_tf32=args.allow_tf32,
|
159 |
device='cuda'
|
|
|
153 |
pipeline_p2p = CatVTONPix2PixPipeline(
|
154 |
base_ckpt=args.p2p_base_model_path,
|
155 |
attn_ckpt=repo_path_mf,
|
156 |
+
attn_ckpt_version="mix-48k-1024",
|
157 |
weight_dtype=init_weight_dtype(args.mixed_precision),
|
158 |
use_tf32=args.allow_tf32,
|
159 |
device='cuda'
|