Update README.md

README.md

@@ -1,25 +1,26 @@
 ---
 license: cc-by-nc-4.0
 language:
-  - en
+- en
+- zh
 tags:
-  - seismology
-  - ground-motion
-  - StyleGAN
-  - earthquake-engineering
-  - generative-model
+- seismology
+- ground-motion
+- StyleGAN
+- earthquake-engineering
+- generative-model
 ---
 
 # UGM Models
 
-[UGM (Utilities for Ground Motions)](../UGM) 的预训练大模型文件仓库。
+[UGM (Utilities for Ground Motions)]的预训练模型文件仓库。
 
 ## 文件
 
 | 文件 | 大小 | 用途 |
 |------|------|------|
-| `KNET-StyleGAN-fid1.730.pkl` | ~309 MB | StyleGAN2 条件生成器（KNET 数据集训练，FID 1.73），用于地震动语谱图生成（`ugm.DGF.time_frequency_generation`） |
-| `vgg16.pt` | ~528 MB | TorchScript 格式的 VGG16 LPIPS 感知损失模型，用于潜空间反演（`ugm.inversion` 的 z / w / w+ 优化） |
+| `KNET-StyleGAN-fid1.730.pkl` | ~309 MB | StyleGAN3 条件生成器（KNET 数据集训练，FID 1.73），用于地震动时频谱生成 |
+| `vgg16.pt` | ~528 MB | TorchScript 格式的 VGG16 LPIPS 感知损失模型，用于潜空间反演 |
 
 ## 下载
 
@@ -32,9 +33,7 @@ python -m ugm.download_models
 python -c "from huggingface_hub import snapshot_download; snapshot_download('xzk8559/UGM', local_dir='~/.cache/ugm/models')"
 ```
 
-模型默认下载到 `~/.cache/ugm/models/`（或 `$UGM_MODEL_DIR`）。
-
 ## 许可
 
 模型权重遵循 **CC BY-NC 4.0**（署名-非商业性使用）。代码仓库见 UGM 主仓库。
-`vgg16.pt` 原始来源为 NVIDIA StyleGAN2 的 LPIPS 权重（NVIDIA 许可）。
+`vgg16.pt` 原始来源为 NVIDIA StyleGAN2 的 LPIPS 权重（NVIDIA 许可）。