[BUG] Colab not working

[zhzLuke96]

          Colab 报错了。。

2024-10-16 14:19:22,235 - numexpr.utils - INFO - NumExpr defaulting to 2 threads.
2024-10-16 14:19:30,519 - root - INFO - new registry table has been added: preprocessor_classes
2024-10-16 14:19:31,422 - root - INFO - new registry table has been added: adaptor_classes
2024-10-16 14:19:31,925 - root - INFO - new registry table has been added: lid_predictor_classes
Traceback (most recent call last):
  File "/content/ChatTTS-Forge/webui.py", line 27, in <module>
    from modules.api.api_setup import process_api_args, setup_api_args
  File "/content/ChatTTS-Forge/modules/api/api_setup.py", line 9, in <module>
    from modules.api.impl import (
  File "/content/ChatTTS-Forge/modules/api/impl/google_api.py", line 16, in <module>
    from modules.core.handler.SSMLHandler import SSMLHandler
  File "/content/ChatTTS-Forge/modules/core/handler/SSMLHandler.py", line 7, in <module>
    from modules.core.pipeline.factory import PipelineFactory
  File "/content/ChatTTS-Forge/modules/core/pipeline/factory.py", line 4, in <module>
    from modules.core.models.zoo.ModelZoo import model_zoo
  File "/content/ChatTTS-Forge/modules/core/models/zoo/__init__.py", line 2, in <module>
    from .ModelZoo import model_zoo
  File "/content/ChatTTS-Forge/modules/core/models/zoo/ModelZoo.py", line 15, in <module>
    from modules.core.models.tts.F5TtsModel import F5TtsModel
  File "/content/ChatTTS-Forge/modules/core/models/tts/F5TtsModel.py", line 15, in <module>
    from modules.repos_static.F5TTS.f5_tts.model import CFM, UNetT, DiT, MMDiT
  File "/content/ChatTTS-Forge/modules/repos_static/F5TTS/f5_tts/model/__init__.py", line 3, in <module>
    from .backbones.unett import UNetT
  File "/content/ChatTTS-Forge/modules/repos_static/F5TTS/f5_tts/model/backbones/unett.py", line 22, in <module>
    from f5_tts.model.modules import (
  File "/content/ChatTTS-Forge/modules/repos_static/F5TTS/f5_tts/model/__init__.py", line 7, in <module>
    from .trainer import Trainer
  File "/content/ChatTTS-Forge/modules/repos_static/F5TTS/f5_tts/model/trainer.py", line 22, in <module>
    from .dataset import DynamicBatchSampler, collate_fn
  File "/content/ChatTTS-Forge/modules/repos_static/F5TTS/f5_tts/model/dataset.py", line 9, in <module>
    from datasets import load_dataset, load_from_disk
ImportError: cannot import name 'load_dataset' from 'datasets' (unknown location)

感觉是相关的

Originally posted by @lei-cao in #176 (comment)

[zhzLuke96]

done de450dd