remove device map

app.py

@@ -23,7 +23,6 @@ def load_model():
             subfolder="text_encoder",
             load_in_8bit=True,
             variant="8bit",
-            device_map="auto",
             token=token
         )
         text_pipe = DiffusionPipeline.from_pretrained(
@@ -31,7 +30,6 @@ def load_model():
             text_encoder=text_encoder,
             unet=None,
             token=token,
-            device_map="auto"  # Add this
         )
         print("Model loaded successfully!")
     return text_pipe