no autocast

app.py

@@ -421,16 +421,15 @@ def process_image(
     print(prompt)
     print(f"\n-------------------------Preprocess done in: {preprocess_time:.2f} seconds-------------------------")    
     start = time.time()
-    with torch.cuda.amp.autocast(True):
-        results = pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            guidance_scale=guidance_scale,
-            num_images_per_prompt=num_images,
-            num_inference_steps=num_steps,
-            generator=generator,
-            image=control_image,
-        ).images[0]
+    results = pipe(
+        prompt=prompt,
+        negative_prompt=negative_prompt,
+        guidance_scale=guidance_scale,
+        num_images_per_prompt=num_images,
+        num_inference_steps=num_steps,
+        generator=generator,
+        image=control_image,
+    ).images[0]
     print(f"\n-------------------------Inference done in: {time.time() - start:.2f} seconds-------------------------")
     # results.save(os.path.join("/data", "temp_image.jpg"))
     # torch.cuda.synchronize()