Update app.py

app.py

@@ -12,16 +12,10 @@ token_hf = os.environ["HF_TOKEN"]
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-base_model_id = "black-forest-labs/FLUX.1-dev"
-repo_name = "ByteDance/Hyper-SD"
-ckpt_name = "Hyper-FLUX.1-dev-8steps-lora.safetensors"
-hyper_lora = hf_hub_download(repo_name, ckpt_name)
-
-pipe = FluxPipeline.from_pretrained(base_model_id, token=token_hf)
-pipe.load_lora_weights(hf_hub_download(repo_name, ckpt_name))
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype)
+pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
 pipe.fuse_lora(lora_scale=0.125)
-pipe.to("cuda", dtype=dtype)
-
+pipe.to(device="cuda", dtype=dtype)
 
 # pipe = FluxPipeline.from_pretrained("sayakpaul/FLUX.1-merged", torch_dtype=torch.bfloat16).to(device)