Views
No views yet
transformers.1import torch
2from transformers import AutoModel
3
4# Загрузка модели
5# Обязательно используйте trust_remote_code=True для активации кастомной архитектуры
6model = AutoModel.from_pretrained("prostochel097/alphapilot-v1", trust_remote_code=True)
7model.eval()
8
9# Симуляция входных данных с датчиков (8 чисел)
10dummy_observation = torch.randn(1, 8)
11
12# Принятие решения
13with torch.no_grad():
14 logits = model(dummy_observation)
15 action = torch.argmax(logits).item()
16
17actions = ["Ничего не делать", "Левый движок", "Главный движок", "Правый движок"]
18print(f"🤖 Решение пилота: {actions[action]}")