From 9c20d449491d464d7635e351e4c2154a35f3ca91 Mon Sep 17 00:00:00 2001
From: wailovet <wailovet@163.com>
Date: Wed, 19 Jun 2024 01:52:49 +0800
Subject: [PATCH] 'update'

---
 mz_train_tools_core.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mz_train_tools_core.py b/mz_train_tools_core.py
index 775b74e..bafd03b 100644
--- a/mz_train_tools_core.py
+++ b/mz_train_tools_core.py
@@ -236,6 +236,7 @@ import logging
 
 
 def generate_toml_config(output_path, enable_bucket=True, resolution=512, batch_size=1, image_dir=None, caption_extension=".caption", num_repeats=10, ):
+    check_install()
     import toml
     config = {
         'general': {