File size: 868 Bytes
9ec3715 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from invert import Inverter
from generate import Generator
from utils import load_config, init_model, seed_everything, get_frame_ids
if __name__ == "__main__":
config = load_config()
pipe, scheduler, model_key = init_model(
config.device, config.sd_version, config.model_key, config.generation.control, config.float_precision)
config.model_key = model_key
seed_everything(config.seed)
print("Start inversion!")
inversion = Inverter(pipe, scheduler, config)
inversion(config.input_path, config.inversion.save_path)
print("Start generation!")
generator = Generator(pipe, scheduler, config)
frame_ids = get_frame_ids(
config.generation.frame_range, config.generation.frame_ids)
generator(config.input_path, config.generation.latents_path,
config.generation.output_path, frame_ids=frame_ids) |