Skip to content

Commit d66afeb

Browse files
authored
fix nas's bug (PaddlePaddle#1019)
1 parent 1a9376a commit d66afeb

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

paddleslim/common/controller_client.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import logging
1818
import socket
1919
from .log_helper import get_logger
20-
20+
import sys
2121
__all__ = ['ControllerClient']
2222

2323
_logger = get_logger(__name__, level=logging.INFO)
@@ -110,7 +110,10 @@ def next_tokens(self):
110110

111111
socket_client.send("next_tokens".encode())
112112
tokens = socket_client.recv(1024).decode()
113-
tokens = [int(token) for token in tokens.strip("\n").split(",")]
113+
try:
114+
tokens = [int(token) for token in tokens.strip("\n").split(",")]
115+
except ValueError:
116+
sys.exit()
114117
return tokens
115118

116119
def request_current_info(self):

0 commit comments

Comments
 (0)