Update app.py

app.py

@@ -72,10 +72,10 @@ else:
     print("CUDA installation not found")
 
 
-device = torch.device('cuda')
+device = 'cuda'
 
 base_model = "black-forest-labs/FLUX.1-dev"
-file_flux = hf_hub_download("marduk191/Flux.1_collection", "flux.1_dev_fp8_fp16t5-marduk191.safetensors")
+file_flux = hf_hub_download("marduk191/Flux.1_collection", "flux.1_dev_8x8_e4m3fn-marduk191.safetensors")
 pipe = FluxPipeline.from_single_file(file_flux, torch_dtype=torch.bfloat16, token=huggingface_token).to(device)
 
 # Load and fuse LoRA BEFORE quantizing