Skip to content

Commit b91cd7b

Browse files
change setting fo random seed to configuration
Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>
1 parent 9ce3a0a commit b91cd7b

File tree

5 files changed

+12
-4
lines changed

5 files changed

+12
-4
lines changed

src/llmcompressor/metrics/logger.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -829,8 +829,8 @@ def __init__(
829829
or SparsificationGroupLogger(
830830
python=log_python,
831831
name=name,
832-
tensorboard=True,
833-
wandb_=True,
832+
tensorboard=False,
833+
wandb_=False,
834834
).loggers
835835
)
836836

tests/lmeval/configs/vl_fp8_dynamic_per_token.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ cadence: weekly
22
model: Qwen/Qwen2-VL-2B-Instruct
33
model_class: TraceableQwen2VLForConditionalGeneration
44
scheme: FP8_DYNAMIC
5+
seed: 42 #compressed model is sensitive to random seed
56
lmeval:
67
model: "hf-multimodal"
78
model_args:

tests/lmeval/configs/vl_int8_w8a8_dynamic_per_token.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ scheme: INT8_dyn_per_token
55
recipe: tests/e2e/vLLM/recipes/INT8/recipe_int8_channel_weight_dynamic_per_token.yaml
66
dataset_id: lmms-lab/flickr30k
77
dataset_split: "test[:512]"
8+
seed: 42 #compressed model is sensitive to random seed
89
lmeval:
910
model: "hf-multimodal"
1011
model_args:

tests/lmeval/configs/vl_w4a16_actorder_weight.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ recipe: tests/e2e/vLLM/recipes/actorder/recipe_w4a16_actorder_weight.yaml
55
dataset_id: lmms-lab/flickr30k
66
dataset_split: "test[:512]"
77
scheme: W4A16_actorder_group
8+
seed: 42 #compressed model is sensitive to random seed
89
lmeval:
910
model: "hf-multimodal"
1011
model_args:

tests/lmeval/test_lmeval.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import os
2+
import random
23
import shutil
34
from pathlib import Path
45

@@ -58,8 +59,6 @@ class TestLMEval:
5859
""" # noqa: E501
5960

6061
def set_up(self):
61-
torch.manual_seed(42)
62-
6362
eval_config = yaml.safe_load(Path(TEST_DATA_FILE).read_text(encoding="utf-8"))
6463

6564
if os.environ.get("CADENCE", "commit") != eval_config.get("cadence"):
@@ -76,6 +75,12 @@ def set_up(self):
7675
self.quant_type = eval_config.get("quant_type")
7776
self.save_dir = eval_config.get("save_dir")
7877

78+
seed = eval_config.get("seed", None)
79+
if seed is not None:
80+
random.seed(seed)
81+
numpy.random.seed(seed)
82+
torch.manual_seed(seed)
83+
7984
logger.info("========== RUNNING ==============")
8085
logger.info(self.scheme)
8186

0 commit comments

Comments
 (0)