Refactor FluxPipeline and AutoencoderKL initialization to remove device assignment

app.py

@@ -20,10 +20,8 @@ from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_imag
 
 llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(device)
-good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16)
 pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
 
 def generate_t2i_prompt(item_name):