diff --git a/src/transformers/commands/serving.py b/src/transformers/commands/serving.py index 801901896d12..b6b5e04c2d59 100644 --- a/src/transformers/commands/serving.py +++ b/src/transformers/commands/serving.py @@ -629,6 +629,13 @@ def run(self): allow_methods=["*"], allow_headers=["*"], ) + logger.warning_once( + "CORS allow origin is set to `*`. This is not recommended for production environments." + ) + else: + logger.warning_once( + "Some apps may require CORS. Consider launching the server with `--enable-cors` if you see errors." + ) @app.post("/v1/chat/completions") def chat_completion(request: dict):