Update app.py

app.py

@@ -34,6 +34,8 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
         return image
 
     if Model == "Club":
+        torch.cuda.max_memory_allocated(device=device)
+        torch.cuda.empty_cache()
         pipe = CogView4Pipeline.from_pretrained("THUDM/CogView4-6B", torch_dtype=torch.bfloat16)
         # Open it for reduce GPU memory usage
         pipe.enable_model_cpu_offload()