Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -136,15 +136,15 @@ 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 |
-
|
145 |
-
|
146 |
-
)
|
147 |
-
pipeline_flux.to("cuda", init_weight_dtype(args.mixed_precision))
|
148 |
|
149 |
|
150 |
|
@@ -369,7 +369,7 @@ def app_gradio():
|
|
369 |
show_type = gr.Radio(
|
370 |
label="Show Type",
|
371 |
choices=["result only", "input & result", "input & mask & result"],
|
372 |
-
value="
|
373 |
)
|
374 |
|
375 |
# num_inference_steps = 50
|
|
|
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 |
|
|
|
369 |
show_type = gr.Radio(
|
370 |
label="Show Type",
|
371 |
choices=["result only", "input & result", "input & mask & result"],
|
372 |
+
value="result only",
|
373 |
)
|
374 |
|
375 |
# num_inference_steps = 50
|