This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| # === Configuration === | |
| SSH_CONFIG="$HOME/.ssh/config" | |
| STATE_FILE="$HOME/.gpucheck_state.json" | |
| LAST_SYNC_FILE="$HOME/.gpucheck_last_sync" | |
| HOSTS_URL="https://vigilant-happiness-o78jgy9.pages.github.io/hosts" | |
| SSH_USER="HDCharles" | |
| POLL_INTERVAL=300 # seconds between checks (5 minutes) | |
| # Three EMA alpha values for different time horizons |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import argparse | |
| import os | |
| import time | |
| import torch | |
| from compressed_tensors.offload import dispatch_model | |
| from compressed_tensors.quantization import preset_name_to_scheme | |
| from datasets import load_dataset | |
| from transformers import AutoModelForCausalLM, AutoTokenizer |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import argparse | |
| import time | |
| import torch | |
| from compressed_tensors.offload import dispatch_model | |
| from compressed_tensors.quantization import preset_name_to_scheme | |
| from datasets import load_dataset | |
| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| from llmcompressor import oneshot |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| """Extract summary data from AWQ DDP regression test log files. | |
| Parses the log output from run_all_tests.sh and produces a comparison table | |
| showing pre-DDP vs post-DDP results across models, schemes, and benchmarks. | |
| Usage: | |
| python extract_log_summary.py regression_results.log | |
| """ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import time, os | |
| model_path = "/mnt/data/engine/HDCharles/hf_hub/models--meta-llama--Meta-Llama-3-8B-Instruct/snapshots/8afb486c1db24fe5011ec46dfbe5b5dccdb575c2" | |
| from vllm import LLM, SamplingParams | |
| print("Loading model...") | |
| t0 = time.time() | |
| llm = LLM(model=model_path, gpu_memory_utilization=0.5, max_logprobs=-1) | |
| tokenizer = llm.get_tokenizer() |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| """Extract summary data from AWQ DDP regression test log files. | |
| Parses the log output from run_all_tests.sh and produces a comparison table | |
| showing pre-DDP vs post-DDP results across models, schemes, and benchmarks. | |
| Usage: | |
| python extract_log_summary.py regression_results.log | |
| """ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| """Extract summary data from AWQ DDP regression test log files. | |
| Parses the log output from run_all_tests.sh and produces a comparison table | |
| showing pre-DDP vs post-DDP results across models, schemes, and benchmarks. | |
| Usage: | |
| python extract_log_summary.py regression_results.log | |
| """ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| """Extract summary data from AWQ DDP regression test log files. | |
| Parses the log output from run_all_tests.sh and produces a comparison table | |
| showing pre-DDP vs post-DDP results across models, schemes, and benchmarks. | |
| Usage: | |
| python extract_log_summary.py regression_results.log | |
| """ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| """Extract summary data from AWQ DDP regression test log files. | |
| Parses the log output from run_all_tests.sh and produces a comparison table | |
| showing pre-DDP vs post-DDP results across models, schemes, and benchmarks. | |
| Usage: | |
| python extract_log_summary.py regression_results.log | |
| """ |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python3 | |
| """Extract summary data from AWQ DDP regression test log files. | |
| Parses the log output from run_all_tests.sh and produces a comparison table | |
| showing pre-DDP vs post-DDP results across models, schemes, and benchmarks. | |
| Usage: | |
| python extract_log_summary.py regression_results.log | |
| """ |
NewerOlder