nvn04 commited on
Commit
3f087a3
·
verified ·
1 Parent(s): 299cbd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -21
app.py CHANGED
@@ -136,28 +136,7 @@ automasker = AutoMasker(
136
  )
137
 
138
 
139
- # Flux-based CatVTON
140
- access_token = os.getenv("HUGGING_FACE_HUB_TOKEN")
141
- flux_repo = "black-forest-labs/FLUX.1-Fill-dev"
142
- pipeline_flux = FluxTryOnPipeline.from_pretrained(flux_repo, use_auth_token=access_token)
143
- pipeline_flux.load_lora_weights(
144
- os.path.join(repo_path, "flux-lora"),
145
- weight_name='pytorch_lora_weights.safetensors'
146
- )
147
- pipeline_flux.to("cuda", init_weight_dtype(args.mixed_precision))
148
-
149
 
150
- # Mask-free CatVTON
151
- catvton_mf_repo = "zhengchong/CatVTON-MaskFree"
152
- repo_path_mf = snapshot_download(repo_id=catvton_mf_repo, use_auth_token=access_token)
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'
160
- )
161
 
162
 
163
  @spaces.GPU(duration=120)
 
136
  )
137
 
138
 
 
 
 
 
 
 
 
 
 
 
139
 
 
 
 
 
 
 
 
 
 
 
 
140
 
141
 
142
  @spaces.GPU(duration=120)