VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

support TI embeds trained by WebUI(?)

Browse Source
This commit is contained in:
Kohya S
2023-03-24 20:46:42 +09:00
parent b2c5b96f2a
commit e203270e31
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gen_img_diffusers.py
View File

@@ -2300,7 +2300,10 @@ def main(args):
else: else:
data = torch.load(embeds_file, map_location="cpu") data = torch.load(embeds_file, map_location="cpu")
if "string_to_param" in data:
data = data["string_to_param"]
embeds = next(iter(data.values())) embeds = next(iter(data.values()))
if type(embeds) != torch.Tensor: if type(embeds) != torch.Tensor:
raise ValueError(f"weight file does not contains Tensor / 重みファイルのデータがTensorではありません: {embeds_file}") raise ValueError(f"weight file does not contains Tensor / 重みファイルのデータがTensorではありません: {embeds_file}")