fix: use identity check for None comparison in model_worker.py

harshadkhetpal · harshadkhetpal · commit 9a384e03bbc8 · 2026-04-12T10:38:22.000+05:30
diff --git a/fastchat/serve/model_worker.py b/fastchat/serve/model_worker.py
@@ -90,7 +90,7 @@ def __init__(
             debug=debug,
         )
         self.device = device
-        if self.tokenizer.pad_token == None:
+        if self.tokenizer.pad_token is None:
             self.tokenizer.pad_token = self.tokenizer.eos_token
         self.context_len = get_context_length(self.model.config)
         self.generate_stream_func = get_generate_stream_function(self.model, model_path)

Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,7 @@ def __init__(`
`90`	`90`	`debug=debug,`
`91`	`91`	`)`
`92`	`92`	`self.device = device`
`93`		`- if self.tokenizer.pad_token == None:`
	`93`	`+ if self.tokenizer.pad_token is None:`
`94`	`94`	`self.tokenizer.pad_token = self.tokenizer.eos_token`
`95`	`95`	`self.context_len = get_context_length(self.model.config)`
`96`	`96`	`self.generate_stream_func = get_generate_stream_function(self.model, model_path)`