# mstool

**Repository Path**: zhangz2001/mstool

## Basic Information

- **Project Name**: mstool
- **Description**: A simple tool for msmodelslim develop.
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 5
- **Created**: 2025-09-02
- **Last Updated**: 2026-01-28

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# MSTool

一个用于模型操作的工具集，提供多种模型相关的功能，包括权重比较等。

## 功能特点

### 权重比较 (cmp)
- 自动查找和比较模型目录中的权重文件
- 支持 safetensors 格式的权重文件
- 提供详细的权重差异分析
- 使用颜色标记突出显示重要差异
- 表格化展示比较结果

## 安装

```bash
# 从源码安装
git clone https://github.com/yourusername/mstool.git
cd mstool
pip install -e .

# 或者直接安装依赖
pip install -r requirements.txt
```

## 使用方法

### 权重比较
```bash
# 使用命令行工具
mstool cmp /path/to/model1 /path/to/model2
```

## 输出说明

权重比较功能会生成一个详细的比较报告，包括：

1. 文件路径信息
2. 键的差异统计
   - 仅在模型1中存在的键
   - 仅在模型2中存在的键
   - 共同存在的键
3. 张量比较结果
   - 形状信息
   - 最大差异（带颜色标记）
   - 平均差异（带颜色标记）

## 依赖要求

- Python >= 3.8
- safetensors >= 0.4.0
- torch >= 2.0.0
- numpy >= 1.24.0
- tabulate >= 0.9.0
- colorama >= 0.4.6

## 许可证

MIT License