ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 11:11:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix some windows dependency issues

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-05 20:28:49 -06:00
parent 8c90fa86c6
commit 64fbd4c92a
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -29,7 +29,9 @@ cd ai-toolkit
git submodule update --init --recursive git submodule update --init --recursive
python3 -m venv venv python3 -m venv venv
source venv/bin/activate 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 pip3 install -r requirements.txt
``` ```

3
requirements.txt
View File

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

4
toolkit/train_tools.py
View File

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