From d6af8d91d6e2f0dd0b2cb7d651273c2fc1f1615e Mon Sep 17 00:00:00 2001 From: mayukhdeb Date: Thu, 14 Mar 2024 01:23:55 -0700 Subject: [PATCH] fix missing imports --- lora_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lora_utils.py b/lora_utils.py index 9ed8c6b..870dc6a 100644 --- a/lora_utils.py +++ b/lora_utils.py @@ -2,8 +2,9 @@ import os, json import torch from safetensors.torch import load_file from typing import Dict - +from peft import PeftModel from dataset_and_utils import TokenEmbeddingsHandler +from safetensors.torch import save_file ''' from diffusers.utils import (