Update app.py

app.py

@@ -12,12 +12,13 @@ import torch
 from openai import OpenAI
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-
+torch.cuda.max_memory_allocated(device=device)
+torch.cuda.empty_cache()
 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()
-
+torch.cuda.empty_cache()
 def clean_string(s):
     s = s.replace("\n", " ")
     s = s.strip()