Update app.py

app.py

@@ -12,6 +12,7 @@ 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)