Slackbot
01/04/2023, 4:30 AMsauyon
01/04/2023, 4:33 AMPhúc Hoàng
01/04/2023, 5:01 AMimport bentoml
from <http://bentoml.io|bentoml.io> import NumpyNdarray
model = bentoml.pytorch.get(
'model:latest').to_runner()
svc = bentoml.Service(
'shrimp-predictor', runners=[model])
@svc.api(input=NumpyNdarray(), output=NumpyNdarray())
async def func(data):
for _ in range(100):
await model.async_run(data)
return data
Phúc Hoàng
01/04/2023, 5:02 AMfrom torch import nn
class Net(nn.Module):
def forward(self, x):
print(f'Model input {x.shape}')
return x
Phúc Hoàng
01/04/2023, 5:02 AMimport bentoml
bentoml.pytorch.save_model('model', Net(), signatures={'__call__': {'batchable': True, 'batch_dim': 0}})
Phúc Hoàng
01/04/2023, 5:02 AMfrom bentoml.client import Client
import numpy as np
bento_client = Client.from_url('<http://localhost:7777>')
data = np.array([[1]])
bento_client.func(data)
Phúc Hoàng
01/04/2023, 5:03 AMpyenv exec python -m bentoml serve --production draft:svc -p 7777
Phúc Hoàng
01/04/2023, 5:03 AMsauyon
01/04/2023, 5:06 AMsauyon
01/04/2023, 5:07 AMcreate_task
instead, and then await those tasks.Phúc Hoàng
01/04/2023, 5:34 AMsauyon
01/04/2023, 5:38 AM