Update merged_files3.py

merged_files3.py

@@ -73,6 +73,15 @@ import numpy as np
 import io
 import base64
 
+if torch.cuda.is_available():
+    torch.backends.cuda.matmul.allow_tf32 = True
+    torch.backends.cudnn.allow_tf32 = True
+    # Set a smaller attention slice size for RTX 2070
+    torch.backends.cuda.max_split_size_mb = 256  # Reduced from 512 for better stability
+    device = torch.device('cuda')
+else:
+    device = torch.device('cpu')
+
 
 MAX_IMAGE_WIDTH = 2048
 IMAGE_FORMAT = "JPEG"