minor fixes

This commit is contained in:
kijai
2024-03-16 18:36:20 +02:00
parent 610dad81b2
commit 15facbbd73
2 changed files with 4 additions and 3 deletions
+1 -1
View File
@@ -3,7 +3,7 @@
.vscode
__pycache__
*.egg-info
*.bak
checkpoints
results
backup
+3 -2
View File
@@ -54,7 +54,8 @@ class DynamiCrafterModelLoader:
if not hasattr(self, 'model') or self.model == None or custom_config != self.current_config:
self.current_config = custom_config
model_path = folder_paths.get_full_path("checkpoints", ckpt_name)
base_name, _ = os.path.splitext(ckpt_name)
ckpt_base_name = os.path.basename(ckpt_name)
base_name, _ = os.path.splitext(ckpt_base_name)
config_file=os.path.join(script_directory, "configs", f"{base_name}.yaml")
config = OmegaConf.load(config_file)
model_config = config.pop("model", OmegaConf.create())
@@ -342,7 +343,7 @@ class DynamiCrafterBatchInterpolation:
video = torch.clamp(video.float(), -1., 1.)
video = (video + 1.0) / 2.0
video = video.squeeze(0).permute(1, 2, 3, 0)
print(f"Sampled {i} / {len(images) - 1}")
print(f"Sampled {i+1} / {len(images) - 1}")
out.append(video)
if not keep_model_loaded: