LTT commited on
Commit
048c43e
·
verified ·
1 Parent(s): 700d917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -197,6 +197,7 @@ def local_normal_global_transform(local_normal_images, azimuths_deg, elevations_
197
  # 生成多视图图像
198
  @spaces.GPU(duration=120)
199
  def generate_multi_view_images(prompt, seed):
 
200
  generator = torch.manual_seed(seed)
201
  with torch.no_grad():
202
  images = flux_pipe(
 
197
  # 生成多视图图像
198
  @spaces.GPU(duration=120)
199
  def generate_multi_view_images(prompt, seed):
200
+ torch.cuda.empty_cache()
201
  generator = torch.manual_seed(seed)
202
  with torch.no_grad():
203
  images = flux_pipe(