mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
refactor(convergence-test): remove unnecessary print (#913)
## Summary <!--- This is a required section; please describe the main purpose of this proposed code change. ---> <!--- ## Details This is an optional section; is there anything specific that reviewers should be aware of? ---> ## Testing Done <!--- This is a required section; please describe how this change was tested. ---> <!-- Replace BLANK with your device type. For example, A100-80G-PCIe Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> - Hardware Type: <BLANK> - [ ] run `make test` to ensure correctness - [ ] run `make checkstyle` to ensure code style - [ ] run `make test-convergence` to ensure convergence
This commit is contained in:
@@ -1212,8 +1212,6 @@ def run_mini_model(
|
||||
else:
|
||||
MINI_MODEL_SETUPS[model_name].liger_kernel_patch_revert_func(**revert_kwargs)
|
||||
|
||||
print(MINI_MODEL_SETUPS[model_name])
|
||||
|
||||
model = create_model(model_name).to(dtype).to(device)
|
||||
|
||||
train_dataset = load_from_disk(DEFAULT_DATASET_PATH)
|
||||
|
||||
Reference in New Issue
Block a user