fix missing imports

2024-03-14 01:23:55 -07:00
parent f83550998e
commit d6af8d91d6
1 changed files with 2 additions and 1 deletions
@@ -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 (