David Jakubec
05/22/2024, 1:11 PMserve failed: BentoCloud API token for default cloud context is required.`
Thought of subprocess to wait for authentication, but it has timed out (probably unreachable).
import subprocess
import xml.etree.ElementTree as ETree
from bentoml.lightgbm import load_model
from pydantic import BaseModel
import bentoml
YOUR_TOKEN = "token"
subprocess.call(f"bentoml cloud login --api-token {YOUR_TOKEN} --endpoint <http://10.244.0.31:7777>".split(" "))
bentoml.models.pull("model-part0:latest")
bentoml.models.pull("model-part1:latest")
bentoml.models.pull("model-part2:latest")
class MessageModel(BaseModel):
message: dict
@bentoml.service(
resources={"cpu": "1"},
traffic={"timeout": 10}
)
class dynamicmodelservice:
def __init__(self):
self.model: list = [
load_model(f'model-part0:latest'),
load_model(f'model-part1:latest'),
load_model(f'model-part2:latest'),
]
@bentoml.api(input_spec=MessageModel)
def predict(self, message: dict) -> dict:
...
Can you please help me? Have already spent so much time on trying to load the model in container.
Thank you very much!Jian Shen Yap
05/22/2024, 4:47 PMDavid Jakubec
05/23/2024, 4:51 AMJian Shen Yap
05/23/2024, 5:01 AMbentoml.cloud sdk does not support yatai unfortunatelyDavid Jakubec
05/23/2024, 5:17 AMbentoml yatai login --api-token {YOUR_TOKEN} --endpoint <http://127.0.0.1:8000>
According to github Yatai ( https://github.com/bentoml/Yatai/blob/main/README.md ) during installation I tried bentoml yatai login and that did not worked. Then I tried bentoml cloud login and it worked just fine