txt2vid fix enumerate(... max_frames)) -> enumerate(... num_steps))

This commit is contained in:
hlky 2022-09-20 14:00:08 +01:00
parent 6009b03752
commit ba84d28da2
No known key found for this signature in database
GPG Key ID: 55A99F1E80D907D5

View File

@ -439,7 +439,7 @@ def txt2vid(
# sample the destination
init2 = torch.randn((1, st.session_state["pipe"].unet.in_channels, height // 8, width // 8), device=torch_device)
for i, t in enumerate(np.linspace(0, 1, max_frames)):
for i, t in enumerate(np.linspace(0, 1, num_steps)):
start = timeit.default_timer()
print(f"COUNT: {frame_index+1}/{max_frames}")