Конвертировать модель. Safetensors в Onnx ⇐ Python

[Anonymous]

Я создал индивидуальную модель с Kohya из модели V1-5-Pruned-Emonly.safetensors и теперь хотел бы импортировать эту модель в приложение C#. Я читал, что я могу использовать ONNX для этого, но я застрял при загрузке весов модели. Код следующим образом
'' '< /p>

Код: Выделить всё

from diffusers import StableDiffusionPipeline
import torch
from safetensors import safe_open

model_path = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)

lora_path = "P718_long_crack_v1.safetensors"

pipe.unet.load_attn_procs(lora_path)

pipe.to("cuda")

unet = pipe.unet
< /code>
'' '
Создание следующей ошибки < /p>
Loading pipeline components...: 100%|██████████| 7/7 [00:01

Подробнее здесь: [url]https://stackoverflow.com/questions/79446020/convert-model-safetensors-in-onnx[/url]