Force this stuff.

2024-09-17 09:59:26 +02:00 · 2024-09-17 09:59:26 +02:00 · a34dbb0ca1
parent c859663f98
commit a34dbb0ca1
1 changed files with 6 additions and 0 deletions
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@ -369,6 +369,12 @@ def launcher(event_loop):
            args.append("--disable-grammar-support")
        if num_shard is not None:
            args.extend(["--num-shard", str(num_shard)])
+        else:
+            try:
+                os.environ["WORLD_SIZE"] = 1
+                del os.environ["NUM_SHARD"]
+            except Exception:
+                pass
        if quantize is not None:
            args.append("--quantize")
            args.append(quantize)