Skip to content

GitLab

Commit 83bc79d6 authored by rprenger's avatar rprenger
Browse files

Remove IP block

parent 9e0ee6fd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
megatron/text_generation_server.py
View file @ 83bc79d6
...@@ -114,8 +114,8 @@ class MegatronGenerate(Resource): ...@@ -114,8 +114,8 @@ class MegatronGenerate(Resource):
if not isinstance(stop_on_eol, bool): if not isinstance(stop_on_eol, bool):
return "stop_on_eol must be a boolean value" return "stop_on_eol must be a boolean value"
if str(request.remote_addr) == "10.14.68.146": # if str(request.remote_addr) == "10.14.68.146":
return "Too many tokens requested from this IP address. Contact Ryan Prenger rprenger@nvidia.com" # return "Too many tokens requested from this IP address. Contact Ryan Prenger rprenger@nvidia.com"
with lock: # Need to get lock to keep multiple threads from hitting code with lock: # Need to get lock to keep multiple threads from hitting code
print("request IP: " + str(request.remote_addr)) print("request IP: " + str(request.remote_addr))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment