more

fzyzcjy · fzyzcjy · commit 52c9e4531905 · 2025-04-09T09:01:53.000+08:00
diff --git a/python/sglang/srt/utils.py b/python/sglang/srt/utils.py
@@ -1796,6 +1796,8 @@ def flatten_nested_list(nested_list):
         return [
             item for sublist in nested_list for item in flatten_nested_list(sublist)
         ]
+    elif isinstance(nested_list, torch.Tensor):
+        return [nested_list.flatten()]
     else:
         return [nested_list]
 

Original file line number	Diff line number	Diff line change
`@@ -1796,6 +1796,8 @@ def flatten_nested_list(nested_list):`
`1796`	`1796`	`return [`
`1797`	`1797`	`item for sublist in nested_list for item in flatten_nested_list(sublist)`
`1798`	`1798`	`]`
	`1799`	`+ elif isinstance(nested_list, torch.Tensor):`
	`1800`	`+ return [nested_list.flatten()]`
`1799`	`1801`	`else:`
`1800`	`1802`	`return [nested_list]`
`1801`	`1803`