tweak

README.md

@@ -29,7 +29,13 @@ pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https:/
 
 Then you can choose from a few options below:
 
-### 1. Local editable
+### 1. As a pip package (if just for inference)
+
+```bash
+pip install git+https://github.com/SWivid/F5-TTS.git
+```
+
+### 2. Local editable (if also do training, finetuning & evaluation)
 
 ```bash
 git clone https://github.com/SWivid/F5-TTS.git
@@ -37,12 +43,6 @@ cd F5-TTS
 pip install -e .
 ```
 
-### 2. As a pip package
-
-```bash
-pip install git+https://github.com/SWivid/F5-TTS.git
-```
-
 ### 3. Build from dockerfile
 ```bash
 docker build -t f5tts:v1 .

src/f5_tts/train/finetune_cli.py

@@ -93,14 +93,14 @@ def main():
         hop_length=hop_length,
     )
 
-    e2tts = CFM(
+    model = CFM(
         transformer=model_cls(**model_cfg, text_num_embeds=vocab_size, mel_dim=n_mel_channels),
         mel_spec_kwargs=mel_spec_kwargs,
         vocab_char_map=vocab_char_map,
     )
 
     trainer = Trainer(
-        e2tts,
+        model,
         args.epochs,
         args.learning_rate,
         num_warmup_updates=args.num_warmup_updates,