Update app.py

app.py

@@ -28,7 +28,7 @@ os.environ["HF_HOME"] = cache_path
 
 torch.backends.cuda.matmul.allow_tf32 = True
 
-pipe = FluxPipeline.from_pretrained("AlekseyCalvin/HSTcolor_FlexSoonr", ignore_mismatched_sizes=True, torch_dtype=torch.bfloat16)
+pipe = FluxPipeline.from_pretrained("AlekseyCalvin/HSTcolor_FlexSoonr", device_map=None, ignore_mismatched_sizes=True, torch_dtype=torch.bfloat16)
 pipe.to(device="cuda", dtype=torch.bfloat16)
 
 model_id = ("zer0int/LongCLIP-GmP-ViT-L-14")