We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a9376a commit d66afebCopy full SHA for d66afeb
paddleslim/common/controller_client.py
@@ -17,7 +17,7 @@
17
import logging
18
import socket
19
from .log_helper import get_logger
20
-
+import sys
21
__all__ = ['ControllerClient']
22
23
_logger = get_logger(__name__, level=logging.INFO)
@@ -110,7 +110,10 @@ def next_tokens(self):
110
111
socket_client.send("next_tokens".encode())
112
tokens = socket_client.recv(1024).decode()
113
- tokens = [int(token) for token in tokens.strip("\n").split(",")]
+ try:
114
+ tokens = [int(token) for token in tokens.strip("\n").split(",")]
115
+ except ValueError:
116
+ sys.exit()
117
return tokens
118
119
def request_current_info(self):
0 commit comments