/
githubmirror
/
Open-Assistant
Обзор
Документация
Войти
/
githubmirror
/
Open-Assistant
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
model/model_training/utils/utils_rl.py
8 строк
259 B
Jin
Add return value in prepare_tensor (#3327)
08 июн 2023, 19:58
Не верифицирован
08 июн 2023, 19:58
960dd2e
Код
Авторство
О чём код?
import tritonclient.grpc as client_util from tritonclient.utils import np_to_triton_dtype def prepare_tensor(name: str, input): t = client_util.InferInput(name, input.shape, np_to_triton_dtype(input.dtype)) t.set_data_from_numpy(input) return t