Update app.py

app.py

@@ -111,7 +111,7 @@ state_dict = torch.load(model_ckpt_path, map_location='cpu')['state_dict']
 state_dict = {k[14:]: v for k, v in state_dict.items() if k.startswith('lrm_generator.')}
 model.load_state_dict(state_dict, strict=True)
 model = model.to(device_1)
-
+torch.cuda.empty_cache()
 @spaces.GPU
 def lrm_reconstructions(image, input_cameras, save_path=None, name="temp", export_texmap=False, if_save_video=False):
     images = image.unsqueeze(0).to(device_1)
@@ -199,7 +199,8 @@ def local_normal_global_transform(local_normal_images, azimuths_deg, elevations_
 @spaces.GPU(duration=120)
 def generate_multi_view_images(prompt, seed):
     # torch.cuda.empty_cache()
-    generator = torch.manual_seed(seed)
+    # generator = torch.manual_seed(seed)
+    generator = torch.Generator().manual_seed(seed)
     with torch.no_grad():
         images = flux_pipe(
             prompt=prompt,