Compare commits

...

2 Commits

Author SHA1 Message Date
DGX 9f9512fa48 movie import fix 2024-02-21 12:43:17 +00:00
DGX 359ef99eaf movie engine fix 2024-02-21 12:42:17 +00:00
1 changed files with 1 additions and 2 deletions

View File

@ -5,7 +5,6 @@ import warnings
import time import time
from tqdm.auto import tqdm from tqdm.auto import tqdm
from PIL import Image from PIL import Image
from latentblending.movie_util import MovieSaver
from typing import List, Optional from typing import List, Optional
import lpips import lpips
import platform import platform
@ -289,7 +288,7 @@ class BlendingEngine():
if t_compute_max_allowed is None and nmb_max_branches is None: if t_compute_max_allowed is None and nmb_max_branches is None:
t_compute_max_allowed = 20 t_compute_max_allowed = 20
elif t_compute_max_allowed is not None and nmb_max_branches is not None: elif t_compute_max_allowed is not None and nmb_max_branches is not None:
raise ValueErorr("Either specify t_compute_max_allowed or nmb_max_branches") raise ValueError("Either specify t_compute_max_allowed or nmb_max_branches")
self.list_idx_injection, self.list_nmb_stems = self.get_time_based_branching(depth_strength, t_compute_max_allowed, nmb_max_branches) self.list_idx_injection, self.list_nmb_stems = self.get_time_based_branching(depth_strength, t_compute_max_allowed, nmb_max_branches)