tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Dan Saunders
2025-09-22 16:24:32 -04:00
parent 8d8fa834a2
commit 61faf4cbe4
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/benchmarks/__init__.py
View File

@@ -1 +1,5 @@
"""Benchmark helpers.""" """Benchmark helpers."""
from .deepseek_v3_moe import DTYPE_MAP, benchmark_deepseek_v3
__all__ = ["benchmark_deepseek_v3", "DTYPE_MAP"]

6
scripts/benchmarks/deepseek_v3_moe_sweep.py
View File

@@ -6,9 +6,15 @@ from __future__ import annotations
import argparse import argparse
import csv import csv
import sys
from pathlib import Path from pathlib import Path
from types import SimpleNamespace from types import SimpleNamespace
CURRENT_DIR = Path(__file__).resolve().parent
PACKAGE_ROOT = CURRENT_DIR.parent
if str(PACKAGE_ROOT) not in sys.path:
sys.path.insert(0, str(PACKAGE_ROOT))
from scripts.benchmarks.deepseek_v3_moe import ( from scripts.benchmarks.deepseek_v3_moe import (
DTYPE_MAP, DTYPE_MAP,
benchmark_deepseek_v3, benchmark_deepseek_v3,