diff --git a/README.md b/README.md
index 9ed568a3..5fd8cf9b 100644
--- a/README.md
+++ b/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
 ```
 
diff --git a/requirements.txt b/requirements.txt
index ca041732..d5697b58 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,4 +13,5 @@ invisible-watermark
 einops
 accelerate
 toml
-albumentations
\ No newline at end of file
+albumentations
+pydantic
\ No newline at end of file
diff --git a/toolkit/train_tools.py b/toolkit/train_tools.py
index 3be84dca..0e18cbc3 100644
--- a/toolkit/train_tools.py
+++ b/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,