diff --git a/mz_train_tools_core_HYDiT.py b/mz_train_tools_core_HYDiT.py index 936a876..3cce23e 100644 --- a/mz_train_tools_core_HYDiT.py +++ b/mz_train_tools_core_HYDiT.py @@ -318,6 +318,13 @@ def check_required(): sys.executable, "-m", "pip", "install", "diffusers" ], check=True) + try: + import transformers + except Exception as e: + subprocess.run([ + sys.executable, "-m", "pip", "install", "transformers" + ], check=True) + def MZ_HYDiTTrain_call(args={}):