Update open_oasis_master/generate.py

open_oasis_master/generate.py

@@ -10,6 +10,7 @@ from utils import one_hot_actions, sigmoid_beta_schedule
 from tqdm import tqdm
 from einops import rearrange
 from torch import autocast
+import os
 #assert torch.cuda.is_available()
 #device = "cuda:0"
 device = "cpu"
@@ -36,9 +37,10 @@ noise_abs_max = 20
 ctx_max_noise_idx = ddim_noise_steps // 10 * 3
 
 # get input video 
+print(os.cwd())
 video_id = "snippy-chartreuse-mastiff-f79998db196d-20220401-224517.chunk_001"
-mp4_path = f"sample_data/{video_id}.mp4"
-actions_path = f"sample_data/{video_id}.actions.pt"
+mp4_path = f"{os.cwd()}/sample_data/{video_id}.mp4"
+actions_path = f"{os.cwd()}/sample_data/{video_id}.actions.pt"
 video = read_video(mp4_path, pts_unit="sec")[0].float() / 255
 actions = one_hot_actions(torch.load(actions_path,map_location=torch.device('cpu')))
 offset = 100