Spaces:
Runtime error
Runtime error
File size: 3,382 Bytes
f1df74a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 |
##### CMCM #####
complex_camera_poses = [
"test_camera_d971457c81bca597",
"test_camera_d971457c81bca597",
"test_camera_d971457c81bca597",
'test_camera_Round-ZoomIn',
'test_camera_Round-ZoomIn',
'test_camera_Round-ZoomIn'
]
complex_camera_pose_prompt = [
"a temple on a mountain, bird's view",
"Effiel Tower in Paris, bird's view",
"a castle in a forest, bird's view",
"a temple on a mountain, bird's view",
"Effiel Tower in Paris, bird's view",
"a castle in a forest, bird's view",
]
basic_camera_poses = [
'test_camera_L',
'test_camera_D',
'test_camera_I',
'test_camera_O',
'test_camera_R',
'test_camera_U',
'test_camera_SPIN-CW-60',
'test_camera_SPIN-ACW-60',
]
basic_camera_pose_prompt = [
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
'coastline, rocks, storm weather, wind, waves, lightning',
]
diff_speeds_camera_poses = [
'test_camera_I_0.2x',
'test_camera_I_0.4x',
'test_camera_I_1.0x',
'test_camera_I_2.0x',
'test_camera_O_0.2x',
'test_camera_O_0.4x',
'test_camera_O_1.0x',
'test_camera_O_2.0x',
]
diff_speeds_camera_pose_prompt = [
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
'A sunrise landscape features mountains and lakes',
]
cmcm_prompt_camerapose = {
'prompts': complex_camera_pose_prompt + basic_camera_pose_prompt + diff_speeds_camera_pose_prompt,
'camera_poses': complex_camera_poses + basic_camera_poses + diff_speeds_camera_poses
}
assert len(cmcm_prompt_camerapose['prompts']) == len(cmcm_prompt_camerapose['camera_poses']), \
"The number of prompts and camera poses should be the same."
### OMCM ###
trajs = [
'shake_1', 'shake_1', 'shake_1',
'curve_2', 'curve_2', 'curve_2',
]
traj_prompt = [
'a sunflower swaying in the wind',
'a rose swaying in the wind',
'a wind chime swaying in the wind',
'a man surfing',
'a man skateboarding',
'a girl skiing'
]
omom_prompt_traj = {
'prompts': traj_prompt,
'trajs': trajs
}
assert len(omom_prompt_traj['prompts']) == len(omom_prompt_traj['trajs']), \
"The number of prompts and trajs should be the same."
both_camerapose = [
'test_camera_O'
]
both_traj = [
'shaking_10'
]
both_prompt = [
'a rose swaying in the wind'
]
both_prompt_camerapose_traj = {
'prompts': both_prompt,
'camera_poses': both_camerapose,
'trajs': both_traj
}
assert len(both_prompt_camerapose_traj['prompts']) == len(both_prompt_camerapose_traj['camera_poses']) == len(both_prompt_camerapose_traj['trajs']), \
"The number of prompts, camera poses and trajs should be the same." |