Request: Can we get this on Colab? about kohya_ss HOT 7 CLOSED

the problem with auto1111 is torch version 1.13.1+cu117. dreambooth only works currently with torch version 1.13.1+cu116

as a gpu solution on the cheap try to find a p40 on ebay. its the pascal titan version with 24gb for enterprise as a compute card and can be had for 200ish dollars on ebay

from kohya_ss.

I wish I had the time... But I don't use colab and would not have the passion to maintain it. If someone create a version I will gladly point to it in the readme.

from kohya_ss.

I wish I could, but I hate Python with a passion, and it would require a colab with a cli.

I tried using the notebook up there with LoRA and 5 hours later everything was an utter failure YET someone I know did fantastic with this on his 3090 YET using the same settings with the notebook I wasted all that time to have nothing to show for it.

from kohya_ss.

I understand. I used to run colab and hated every minutes of it... Hence why I invested in a 3090 so I could free myself from remote GPUs.

from kohya_ss.

I understand. I used to run colab and hated every minutes of it... Hence why I invested in a 3090 so I could free myself from remote GPUs.

They insulate us far too much, and I feel like I am in a box when I use them, plus every minute I feel extremely rushed because my 2.5h per day is ticking away. Another thing is to help drain the time away I have to constantly redownload, and update, everything (seems all notebooks do this) so by the time it actually starts I wasted 10 minutes right there.

Considering what Nvidia has done with their utter greed my saving for what was supposed to be the 16GB 4060 for MSRP $379 was destroyed so when that happened I went to colab. With everything that has happened, and all the notebooks being mostly, to completely, broken, I have almost lost all interest in this stuff now.

from kohya_ss.

Alrighty, then how about an Automatic1111 webgui extension? If it is an extension I can run that on Colab (everyone can), and you need not worry about colab at all (as all my other extensions)? The Dreambooth LoRA trainer is not nearly as good AND that dev went nuts and started closing tickets, and telling his users off when they submitted a ticket as well as letting the world know some bad language stuff. I bet it could be done considering both are webgui gradio based.

from kohya_ss.

i hope you can do this i try to do but i have many errors but the base maybe work for do this implementation i share the code

!git clone https://github.com/bmaltais/kohya_ss.git
%cd kohya_ss

!pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
!pip install --use-pep517 --upgrade -r requirements.txt
!pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl

!wget -N https://github.com/korakot/kora/releases/download/v0.10/py310.sh
!bash ./py310.sh -b -f -p /usr/local
!python -m ipykernel install --name "py310" --user
!conda install -c "xformers/label/dev" xformers -y
!pip install open-clip-torch
!pip install triton

!pip install tensorflow-gpu==2.3.0
!pip install tensorflow-compression
!pip install tensorrt

!pip install --use-pep517 --upgrade -r requirements.txt

%cd /content/sample_data/kohya_ss/venv/lib/python3.10/site-packages
!pip install accelerate
!conda install -c conda-forge accelerate
!pip install git+https://github.com/huggingface/accelerate
!git clone https://github.com/huggingface/accelerate
%cd accelerate
!pip install -e .
#!accelerate config
!python -c "from accelerate.utils import write_basic_config; write_basic_config(mixed_precision='fp16')"
!accelerate env

import gradio as gr
import os
import argparse
from dreambooth_gui import dreambooth_tab
from finetune_gui import finetune_tab
from textual_inversion_gui import ti_tab
from library.utilities import utilities_tab
from library.extract_lora_gui import gradio_extract_lora_tab
from library.merge_lora_gui import gradio_merge_lora_tab
from lora_gui import lora_tab

def UI(username, password):

css = ''

if os.path.exists('./style.css'):
    with open(os.path.join('./style.css'), 'r', encoding='utf8') as file:
        print('Load CSS...')
        css += file.read() + '\n'

interface = gr.Blocks(css=css, title="Kohya_ss GUI")

with interface:
    with gr.Tab('Dreambooth'):
        (
            train_data_dir_input,
            reg_data_dir_input,
            output_dir_input,
            logging_dir_input,
        ) = dreambooth_tab()
    with gr.Tab('Dreambooth LoRA'):
        lora_tab()
    with gr.Tab('Dreambooth TI'):
        ti_tab()
    with gr.Tab('Finetune'):
        finetune_tab()
    with gr.Tab('Utilities'):
        utilities_tab(
            train_data_dir_input=train_data_dir_input,
            reg_data_dir_input=reg_data_dir_input,
            output_dir_input=output_dir_input,
            logging_dir_input=logging_dir_input,
            enable_copy_info_button=True,
        )
        gradio_extract_lora_tab()
        gradio_merge_lora_tab()

# Show the interface
if not username == '':
    interface.launch(auth=(username, password))
else:
    interface.launch(debug=True, share=True)

if name == 'main':
# torch.cuda.set_per_process_memory_fraction(0.48)
UI(username='', password='')
parser.add_argument(
'--username', type=str, default='', help='Username for authentication'
)
parser.add_argument(
'--password', type=str, default='', help='Password for authentication'
)

args = parser.parse_args()

UI(username=args.username, password=args.password)

from kohya_ss.