stzhao commited on
Commit
a66df57
·
verified ·
1 Parent(s): 16ff32e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -86,7 +86,7 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
86
  system_prompt="You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.",
87
  ).images[0]
88
 
89
- torch.cuda.empty_cache()
90
 
91
  return image
92
 
 
86
  system_prompt="You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.",
87
  ).images[0]
88
 
89
+ # torch.cuda.empty_cache()
90
 
91
  return image
92