ge3

main.py

@@ -22,7 +22,7 @@ try:
         MODEL_NAME,
         torch_dtype=torch.float32,
         device_map="auto",
-        low_cpu_mem_usage=False
+        low_cpu_mem_usage=True
     )
 
     generator = pipeline(