From abead52a2fb90f965b5d17bb77cf45366c247766 Mon Sep 17 00:00:00 2001 From: mayukhdeb Date: Sun, 31 Mar 2024 04:47:26 -0700 Subject: [PATCH] fix import --- trainer/utils/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trainer/utils/inference.py b/trainer/utils/inference.py index 6b0fa0a..819f558 100644 --- a/trainer/utils/inference.py +++ b/trainer/utils/inference.py @@ -6,7 +6,7 @@ import json import gc from diffusers import EulerDiscreteScheduler from .val_prompts import val_prompts -from .models import load_models +from ..models import load_models from .lora import patch_pipe_with_lora from .prompt import prepare_prompt_for_lora from .io import make_validation_img_grid