Merge branch 'main' of github.com:ostris/ai-toolkit

README.md

@@ -29,7 +29,9 @@ cd ai-toolkit
 git submodule update --init --recursive
 python3 -m venv venv
 source venv/bin/activate
-# or source venv/Scripts/activate on windows
+# .\venv\Scripts\activate on windows
+# windows install pytorch first with 
+# pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu117
 pip3 install -r requirements.txt
 ```
 

requirements.txt

@@ -13,4 +13,5 @@ invisible-watermark
 einops
 accelerate
 toml
-albumentations
+albumentations
+pydantic

toolkit/train_tools.py

@@ -4,6 +4,10 @@ import json
 import os
 import time
 from typing import TYPE_CHECKING
+import sys
+from toolkit.paths import SD_SCRIPTS_ROOT
+
+sys.path.append(SD_SCRIPTS_ROOT)
 
 from diffusers import (
     StableDiffusionPipeline,