zhengchong commited on
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.

Files changed (1) hide show
  1. app.py +1 -1
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'