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

Files changed (1) hide show
  1. app.py +2 -2
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=repo_path,
155
- attn_ckpt_version="mix-48k-1024",
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'