[serving] Fix typo
This commit is contained in:
@@ -14,14 +14,14 @@ try:
|
|||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from starlette.responses import JSONResponse
|
from starlette.responses import JSONResponse
|
||||||
|
|
||||||
_serve_dependancies_installed = True
|
_serve_dependencies_installed = True
|
||||||
except (ImportError, AttributeError):
|
except (ImportError, AttributeError):
|
||||||
BaseModel = object
|
BaseModel = object
|
||||||
|
|
||||||
def Body(*x, **y):
|
def Body(*x, **y):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
_serve_dependancies_installed = False
|
_serve_dependencies_installed = False
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger("transformers-cli/serving")
|
logger = logging.getLogger("transformers-cli/serving")
|
||||||
@@ -111,7 +111,7 @@ class ServeCommand(BaseTransformersCLICommand):
|
|||||||
self.port = port
|
self.port = port
|
||||||
self.workers = workers
|
self.workers = workers
|
||||||
|
|
||||||
if not _serve_dependancies_installed:
|
if not _serve_dependencies_installed:
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
"Using serve command requires FastAPI and unicorn. "
|
"Using serve command requires FastAPI and unicorn. "
|
||||||
'Please install transformers with [serving]: pip install "transformers[serving]".'
|
'Please install transformers with [serving]: pip install "transformers[serving]".'
|
||||||
|
|||||||
Reference in New Issue
Block a user