github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fixup relative path to devices module

Browse Source
This commit is contained in:
Lincoln Stein
2023-04-10 23:44:58 -04:00
parent 0b22a3f34d
commit 2251d3abfe
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ldm/modules/kohya_lora_manager.py
View File

@@ -7,7 +7,8 @@ from safetensors.torch import load_file
from torch.utils.hooks import RemovableHandle
from transformers import CLIPTextModel
from ldm.invoke.devices import choose_torch_device
from ..invoke.devices import choose_torch_device
"""
This module supports loading LoRA weights trained with https://github.com/kohya-ss/sd-scripts