FLUXPro

Running on Zero

KingNish commited on Sep 19

Commit

deb928c

•

1 Parent(s): c82ffd4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ dtype = torch.float16
 pipe = FLUXPipelineWithIntermediateOutputs.from_pretrained(
     "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
 ).to("cuda")
-pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16)
 torch.cuda.empty_cache()
 # Inference function

 pipe = FLUXPipelineWithIntermediateOutputs.from_pretrained(
     "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
 ).to("cuda")
+pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 torch.cuda.empty_cache()
 # Inference function