@BezdomnyiBox
Студент программной инженерии

Как написать callback для progress bar?

Есть генеративная модель runwayml/stable-diffusion-v1-5, хочу чтобы через websocket передавался на index.html прогресс генерации изображения.

Подскажите, как надо написать progress_callback?

num_inference_steps = 25  # Вы можете настроить это значение

    
    def progress_callback(pipeline, step_index, timestep, callback_kwargs):
        progress = int((step_index + 1) / num_inference_steps * 100)
        socketio.emit('progress', {'progress': progress})
    
    # Генерация изображения
    with torch.no_grad():
        image = pipeline(
            text_prompt,
            num_inference_steps=num_inference_steps,
            callback_on_step_end=progress_callback
        ).images[0]
  • Вопрос задан
  • 87 просмотров
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Похожие вопросы