Skip to content

Commit

Permalink
docs: Fix cpu as strings (#4735)
Browse files Browse the repository at this point in the history
Fix cpu as strings

Signed-off-by: Sherlock113 <[email protected]>
  • Loading branch information
Sherlock113 authored May 16, 2024
1 parent 3c33b4d commit 7d380d2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/source/guides/model-composition.rst
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ This example defines a sequential pipeline where output from one model is fed as
import numpy as np
from transformers import pipeline
@bentoml.service(resources={"cpu": 2, "memory": "2Gi"})
@bentoml.service(resources={"cpu": "2", "memory": "2Gi"})
class PreprocessingService:
model_a_ref = bentoml.models.get("model_a:latest")
Expand Down Expand Up @@ -253,7 +253,7 @@ This example runs two independent models concurrently to generate different type
data_b = self.pipeline_b(input_data)
return data_b
@bentoml.service(resources={"cpu": 4, "memory": "8Gi"})
@bentoml.service(resources={"cpu": "4", "memory": "8Gi"})
class EnsembleService:
service_a = bentoml.depends(ModelAService)
service_b = bentoml.depends(ModelBService)
Expand Down

0 comments on commit 7d380d2

Please sign in to comment.