Извлечь триггерные слова из файла SafetensorsPython

Программы на Python
Anonymous
Извлечь триггерные слова из файла Safetensors

Сообщение Anonymous »


Во многих файлах .safetensors, связанных с разными авторами LoRA, иногда указываются «триггерные слова». Я пытался извлечь их из файла .safetensors, но не могу их найти. Их нет в разделе метаданных, и согласно документации я не могу придумать другого места, где они могут быть.

С другой стороны, я знаю, что эти триггерные слова работают, потому что при их предоставлении я получаю изображение.

Я думал, что сработает что-то вроде этого: из защитных тензоров импортируйте Safe_open

тензоры = {} с Safe_open("lora.safetensors", framework="pt") в качестве f: печать (f.metadata()) Но, похоже, доступ сложнее. Есть предложения?

Вернуться в «Python»