Boundaries now load on cpu

generate_videos.py

@@ -113,7 +113,7 @@ def interpolate_with_boundaries(args, source_latent, alphas):
     boundaries_and_distances = []
     for direction_type in edit_directions:
         distances = SUGGESTED_DISTANCES[direction_type]
-        boundary = torch.load(os.path.join(boundary_dir, f'{direction_type}.pt')).cpu().detach().numpy()
+        boundary = torch.load(os.path.join(boundary_dir, f'{direction_type}.pt'), map_location="cpu").numpy()
 
         for distance in distances:
             if distance: