Skip to content

[Bug]: Triton module not found error on MacOS #1182

@andrerivas

Description

@andrerivas

What happened?

Did #1034 break MacOS? I'm seeing this on Mac after installing and trying to start:

➜  OneTrainer git:(master) ./start-ui.sh
[OneTrainer] Using Python Venv environment in "venv"...
[OneTrainer] + python scripts/util/version_check.py 3.10 3.13
[OneTrainer] + python scripts/train_ui.py
Traceback (most recent call last):
  File "/Users/user/OneTrainer/scripts/train_ui.py", line 5, in <module>
    from modules.ui.TrainUI import TrainUI
  File "/Users/user/OneTrainer/modules/ui/TrainUI.py", line 22, in <module>
    from modules.ui.ConvertModelUI import ConvertModelUI
  File "/Users/user/OneTrainer/modules/ui/ConvertModelUI.py", line 5, in <module>
    from modules.util import create
  File "/Users/user/OneTrainer/modules/util/create.py", line 6, in <module>
    from modules.dataLoader.ChromaBaseDataLoader import ChromaBaseDataLoader
  File "/Users/user/OneTrainer/modules/dataLoader/ChromaBaseDataLoader.py", line 6, in <module>
    from modules.model.ChromaModel import ChromaModel
  File "/Users/user/OneTrainer/modules/model/ChromaModel.py", line 4, in <module>
    from modules.model.BaseModel import BaseModel, BaseModelEmbedding
  File "/Users/user/OneTrainer/modules/model/BaseModel.py", line 6, in <module>
    from modules.module.LoRAModule import LoRAModuleWrapper
  File "/Users/user/OneTrainer/modules/module/LoRAModule.py", line 13, in <module>
    from modules.util.quantization_util import get_unquantized_weight, get_weight_shape
  File "/Users/user/OneTrainer/modules/util/quantization_util.py", line 78, in <module>
    from modules.module.quantized.LinearGGUFA8 import LinearGGUFA8
  File "/Users/user/OneTrainer/modules/module/quantized/LinearGGUFA8.py", line 5, in <module>
    from modules.util.triton_mm_8bit import mm_8bit as triton_mm_8bit
  File "/Users/user/OneTrainer/modules/util/triton_mm_8bit.py", line 16, in <module>
    import triton
ModuleNotFoundError: No module named 'triton'

What did you expect would happen?

The app should launch.

Relevant log output

Generate and upload debug_report.log

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions