Update app.py

app.py

@@ -81,8 +81,10 @@ model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
 pipe = DiffusionPipeline.from_pretrained(
     LOCAL_TURBOX,
     trust_remote_code=True,
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
     local_files_only=True,
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+    low_cpu_mem_usage=False,
+    device_map=None
 )
 pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", local_files_only=True, trust_remote_code = True, shift=5)
 pipe = pipe.to(device)