Update app.py

app.py

@@ -35,12 +35,11 @@ print(device)
 base_model = "black-forest-labs/FLUX.1-dev"
 
 # load pipe
-taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
-good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device)
+taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to_empty(device)
+good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to_empty(device)
 
-txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to(device)
+txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to_empty(device)
 txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
-txt2img_pipe.to(device)
 
 
 MAX_SEED = 2**32 - 1