Upload app.py

app.py

@@ -19,7 +19,7 @@ transformer = FluxTransformer2DModel.from_pretrained(flux_repo, subfolder="trans
 nf4_quantization_config = BitsAndBytesConfig(load_in_4bit=True)
 transformer_nf4 = FluxTransformer2DModel.from_pretrained(flux_repo, subfolder="transformer", quantization_config=nf4_quantization_config,
                                                          torch_dtype=torch.bfloat16, token=HF_TOKEN)
-pipe = FluxPipeline.from_pretrained(flux_repo, transformer=transformer, torch_dtype=torch.bfloat16, token=HF_TOKEN).to("cpu")
+pipe = FluxPipeline.from_pretrained(flux_repo, transformer=transformer, torch_dtype=torch.bfloat16, token=HF_TOKEN)
 hyper_sd_lora = hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors")
 
 @spaces.GPU(duration=70)