move pipeline to gpu

app.py

@@ -25,13 +25,14 @@ def load_model():
             load_in_8bit=True,
             variant="8bit",
             token=token
-        ).to(device)
+        )
         text_pipe = DiffusionPipeline.from_pretrained(
             "DeepFloyd/IF-I-L-v1.0",
             text_encoder=text_encoder,
             unet=None,
             token=token,
         )
+        text_pipe = text_pipe.to(device)
         print("Model loaded successfully!")
     return text_pipe