# model-play

**Repository Path**: feifei14119/model-play

## Basic Information

- **Project Name**: model-play
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-03-07
- **Last Updated**: 2026-04-27

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# model-play

本地模型部署与信息查看脚本（支持 ROCm/AMD GPU 环境）。

## model_info.py — 通用模型信息

`model_info.py` 可输出**任意路径**下模型的信息：配置摘要、模块层级、参数量。自动识别 HuggingFace 风格（含 `config.json`）与 Hunyuan3D 风格（子目录含 `config.yaml` / `model.fp16.safetensors`）。

### 基本用法

```bash
# 指定模型目录（自动识别类型）
python3 model_info.py /path/to/model
```

### HuggingFace 风格模型

适用于 Qwen3-0.6B、Qwen3-TTS、以及其它含 `config.json` 的模型：

```bash
python3 model_info.py /feifei/repo/Qwen3-0.6B
python3 model_info.py /feifei/repo/Qwen3-TTS-0.6B
```

- **仅看配置与权重摘要、不加载模型**（省内存，避免 ROCm 下段错误）：

  ```bash
  python3 model_info.py /path/to/model --no-load
  ```

- 使用 GPU：`python3 model_info.py /path/to/model --device cuda`

### Hunyuan3D 风格模型

根目录下存在带 `config.yaml` 或 `model.fp16.safetensors` 的子目录时，按 Hunyuan3D 处理：

```bash
# 列出可用子目录
python3 model_info.py /feifei/repo/Hunyuan3D-2mini --list

# 指定子目录，通过 pipeline 加载并打印层级
python3 model_info.py /feifei/repo/Hunyuan3D-2mini -s hunyuan3d-dit-v2-mini

# 仅从 safetensors 解析（无需 hy3dgen / GPU）
python3 model_info.py /feifei/repo/Hunyuan3D-2mini -s hunyuan3d-dit-v2-mini --safetensors-only
```

### 参数说明

| 参数 | 说明 |
|------|------|
| `model_path` | 模型目录路径（必填） |
| `-s`, `--subfolder` | (Hunyuan3D) 子目录名 |
| `--safetensors-only` | (Hunyuan3D) 只解析 safetensors，不加载 pipeline |
| `--list` | (Hunyuan3D) 只列出子目录后退出 |
| `--device` | 加载设备，默认 `cpu` |
| `--no-load` | (HuggingFace) 不加载模型，仅打印 config 与权重摘要 |

---

## run_qwen3_0.6b.py — Qwen3-0.6B 文本生成

本地运行 **Qwen3-0.6B** 小语言模型，支持单条提示或交互式对话。适用于 ff_rocm7 等容器（含 ROCm 时脚本会自动规避 torchvision 兼容问题）。

### 基本用法

```bash
# 单条 prompt，跑完即退出（默认 CPU）
python3 run_qwen3_0.6b.py --prompt "你好，请用一句话介绍你自己。"

# 指定模型路径
python3 run_qwen3_0.6b.py --model_path /feifei/repo/Qwen3-0.6B --prompt "写一首短诗"

# 使用 GPU
python3 run_qwen3_0.6b.py --prompt "你好" --device cuda
```

### 交互模式

不传 `--prompt` 时进入逐轮对话，输入内容回车生成，空行退出：

```bash
python3 run_qwen3_0.6b.py
```

### 常用参数

| 参数 | 说明 | 默认 |
|------|------|------|
| `--model_path` | 模型目录 | `$MODEL_PATH` 或 `/feifei/repo/Qwen3-0.6B` |
| `--prompt` | 单条提示（不指定则交互） | 无 |
| `--device` | `cpu` 或 `cuda` | `cpu` |
| `--max_new_tokens` | 最大生成 token 数 | 256 |
| `--temperature` | 采样温度，0 为贪婪 | 0.7 |

---

## run_qwen_tts_cpu.py — Qwen3-TTS 0.6B 语音合成（CPU）

在 **CPU** 上运行 **Qwen3-TTS-0.6B**，将文本合成为语音。脚本内已做 ROCm/核显相关修复（如 speaker encoder），适合在 ff_rocm7 等环境用 CPU 跑，避免 GPU 段错误。

### 基本用法

```bash
# 直接运行（使用脚本内默认文本，输出到当前目录）
python3 run_qwen_tts_cpu.py
```

默认将合成结果保存为 **`qwen_tts_amd_output.wav`**（脚本内写死路径，可在代码中改）。

### 可选环境变量

- **`REF_AUDIO`**：参考音频路径（用于音色克隆）；不设则用静音作为默认参考。
- **`REF_TEXT`**：参考音频对应文本；不设则用空格。

```bash
REF_AUDIO=/path/to/ref.wav REF_TEXT="参考音频里的文字" python3 run_qwen_tts_cpu.py
```

### 说明

- 模型路径在脚本内写死为 `/feifei/repo/Qwen3-TTS-0.6B`，需先下载好模型。
- 合成文本在脚本内写死（约「你好，这是 AMD 核显运行…」），要改内容需编辑脚本中的 `text` 变量。
- 建议文本长度 ≤200 字，以控制显存/内存与速度。

---

## run_hunyuan3d_mini.py — Hunyuan3D-2mini 图像转 3D

使用 **Hunyuan3D-2mini** 将单张图片转为 3D mesh（.glb）。需先部署好模型并安装 hy3dgen（参见 `deploy_hunyuan3d.sh`）。

### 基本用法

```bash
# 指定输入图片，输出为同名的 .glb（默认 CPU）
python3 run_hunyuan3d_mini.py --image /path/to/photo.png

# 指定输出路径
python3 run_hunyuan3d_mini.py --image photo.jpg --output result.glb

# 使用 GPU 加速
python3 run_hunyuan3d_mini.py --image photo.png --device cuda
```

### 常用参数

| 参数 | 说明 | 默认 |
|------|------|------|
| `--image` | 输入图片路径（必填） | — |
| `--model-path` | 模型根目录 | `/feifei/repo/Hunyuan3D-2mini` |
| `--subfolder` | 模型子目录 | `hunyuan3d-dit-v2-mini` |
| `--output` | 输出 .glb 路径 | 与输入同名的 .glb |
| `--device` | `cpu` 或 `cuda` | `cpu` |
| `--num-inference-steps` | 推理步数 | 30 |
| `--octree-resolution` | 八叉树分辨率 | 380 |
| `--num-chunks` | chunk 数量 | 20000 |
| `--seed` | 随机种子 | 12345 |

---

## 模型下载示例

```bash
modelscope download --model Qwen/Qwen3-TTS-12Hz-0.6B-Base --local_dir /feifei/repo/Qwen3-TTS
modelscope download --model Tencent-Hunyuan/Hunyuan3D-2mini
```