removed device map.

app.py

@@ -78,7 +78,7 @@ def load_pipeline_single_gpu():
             trust_remote_code=True,
             torch_dtype=dtype,
             low_cpu_mem_usage=True,
-            device_map="balanced_low_0",
+            # device_map="balanced_low_0",
             offload_folder=offload_dir,
             offload_state_dict=True,
             # max_memory=max_memory,
@@ -102,7 +102,7 @@ def load_pipeline_single_gpu():
             slider_projector=slider_projector,
             torch_dtype=weight_dtype,
             low_cpu_mem_usage=True,
-            device_map="balanced_low_0",
+            # device_map="balanced_low_0",
             offload_folder=offload_dir,
             offload_state_dict=True,
             # max_memory=max_memory,