Update app.py

app.py

@@ -14,7 +14,9 @@ from openai import OpenAI
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = CogView4Pipeline.from_pretrained("THUDM/CogView4-6B", torch_dtype=torch.bfloat16).to(device)
-
+pipe.enable_model_cpu_offload()
+pipe.vae.enable_slicing()
+pipe.vae.enable_tiling()
 
 def clean_string(s):
     s = s.replace("\n", " ")