Проблема с загрузкой компонентов FluxPipeline. ⇐ Python

[Anonymous]

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

import torch
from diffusers import FluxPipeline

pipe = FluxPipeline.from_pretrained('C:\\Python\\Projects\\test1\\flux1dev', torch_dtype=torch.bfloat16)
pipe.enable_sequential_cpu_offload()

prompt = "beach ball"
image = pipe(
prompt,
height = 1024,
width = 1024,
guidance_scale = 3.5,
num_inference_steps = 50,
max_sequence_length = 512,
generator = torch.Generator("cpu").manual_seed(0)
).images[0]
image.save("beach ball.png")

Я столкнулся с проблемой при выполнении этого простого теста Flux.1.
Каждый раз, когда я пытался запустить код, он просто останавливался после загрузки части. компонентов конвейера.
Ни одного исключения или кода ошибки не было выдано, и никаких выходных данных не было, программа просто остановлена.
Я действительно понятия не имею, что Я делаю, и я просто возился с Flux, чтобы посмотреть, что он может do.
Есть также очень странная проблема в терминале, когда последнее загружаемое содержимое помещается в следующую строку терминала.

Подробнее здесь: https://stackoverflow.com/questions/788 ... components