diff --git a/utils.py b/utils.py index 34a1bb3..1f93da6 100644 --- a/utils.py +++ b/utils.py @@ -41,8 +41,8 @@ def load_model_from_config(config, ckpt, verbose=False, device='cuda'): def prepare_default_model(default_seed = 42): - default_config_path = "/home/subrtade/analogies/DiffusionImageAnalogies/stable-diffusion/configs/stable-diffusion/v1-inference.yaml" - default_ckpt_path = "/home/subrtade/analogies/DiffusionImageAnalogies/stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt" + default_config_path = "configs/stable-diffusion/v1-inference.yaml" + default_ckpt_path = "models/ldm/stable-diffusion-v1/model.ckpt" seed_everything(default_seed) @@ -267,4 +267,4 @@ def check_and_run_inversion(model, file_id, subfolder, config, tokens = True): export_path = os.path.join(config.path2save_prefix, file_id, subfolder) save_results2pickle(export_path, output) - print(f'Inversion done') \ No newline at end of file + print(f'Inversion done')