Views
No views yet
.safetensors (безопасный и быстрый)1from safetensors.torch import load_file
2import torch
3
4# Загрузка весов
5state_dict = load_file("model.safetensors")
6
7# Инициализация модели (с теми же параметрами, что и при обучении)
8model = VideoGenerator(vocab_size=10)
9
10# Загрузка весов
11model.load_state_dict(state_dict)
12model.eval()
13
14# Генерация видео по промпту
15# (предварительно токенизируй текст с помощью vocab.json)Фигуры: квадраты
Цвета: красный, зелёный, синий, жёлтый
Движения: перемещение, отскокНизкое разрешение (64×64)
Простые геометрические объекты
Короткая длительность (8 кадров)
Не поддерживает сложные сцены