Spaces:

gokaygokay
/

FLUX.1-dev-with-Captioner

Running on Zero

gokaygokay commited on Oct 29, 2024

Commit

57b54a6

verified ·

1 Parent(s): e093f64

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,6 +36,7 @@ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=
 pipe.load_lora_weights(hf_hub_download("gokaygokay/Flux-Game-Assets-LoRA-v2", "game_asst.safetensors"))
 pipe.fuse_lora(lora_scale=1)
 pipe.to(device="cuda", dtype=torch.bfloat16)
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(

 pipe.load_lora_weights(hf_hub_download("gokaygokay/Flux-Game-Assets-LoRA-v2", "game_asst.safetensors"))
 pipe.fuse_lora(lora_scale=1)
 pipe.to(device="cuda", dtype=torch.bfloat16)
+pipe.enable_model_cpu_offload()
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(