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