Ядро Jupyter аварийно завершает работу/перезагружается при загрузке локальной модели HuggingFace .safetensors с помощью Python

Программы на Python
Anonymous
Ядро Jupyter аварийно завершает работу/перезагружается при загрузке локальной модели HuggingFace .safetensors с помощью

Сообщение Anonymous »

Я пытаюсь загрузить большую локальную модель Hugging Face (
./ltx-2-19b-dev.safetensors", формат .safetensors) внутри JupyterLab с помощью преобразователей, но ядро ​​сразу же аварийно завершает работу во время from_pretrained() без какой-либо обратной трассировки Python.

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

When I run this code:

from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "./ltx-2-19b-dev.safetensors"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)
Ядро немедленно перезапускается, и появляется всплывающее окно с сообщением чего-то вроде:

Похоже, ядро ​​для Untitled.ipynb умерло. Он перезапустится автоматически.

В записной книжке не отображается обратная трассировка Python — просто происходит сбой.
Почему AutoModelForCausalLM.from_pretrained() может привести к сбою ядра без обратной трассировки?

Подробнее здесь: https://stackoverflow.com/questions/798 ... nsors-mode

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