Skip to content

Commit

Permalink
delete real path
Browse files Browse the repository at this point in the history
  • Loading branch information
StarConnor committed Jun 29, 2024
1 parent 20a432d commit b85c98e
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,14 +8,13 @@ def pytest_addoption(parser):
parser.addoption("--layer", nargs="*", type=int, required=False, help='Layer number')
parser.addoption("--batch_size", type=int, required=False, default=4096, help='Batchsize, default 4096')
parser.addoption("--lr", type=float, required=False, default=8e-5, help='Learning rate, default 8e-5')
parser.addoption("--expdir", type=str, required=False, default="/remote-home/fkzhu/zfk/engineering/test/Language-Model-SAEs/results", help='Export directory, default zfk/ftresults_KL')
parser.addoption("--expdir", type=str, required=False, default="path/to/results", help='Export directory, default path')
parser.addoption("--useddp", type=bool, required=False, default=False, help='If using distributed method, default False')
parser.addoption('--attn_type', type=str, required=False, choices=['flash', 'normal'], default="flash", help='Use or not use log of wandb, default True')
parser.addoption('--dtype', type=str, required=False, choices=['fp32', 'bfp16'], default="fp32", help='Dtype, default fp32')
parser.addoption('--model_name', type=str, required=False, default="meta-llama/Meta-Llama-3-8B", help='Supported model name of TransformerLens, default gpt2')
parser.addoption('--d_model', type=int, required=False, default=4096, help='Dimension of model hidden states, default 4096')
# FIXME remove default model path
parser.addoption('--model_path', type=str, required=False, default="/remote-home/share/models/llama3_hf/Meta-Llama-3-8B", help='Hugging-face model path used to load.')
parser.addoption('--model_path', type=str, required=False, default="path/to/model", help='Hugging-face model path used to load.')

@pytest.fixture
def args(request):
Expand Down

0 comments on commit b85c98e

Please sign in to comment.