mirror of https://github.com/bmaltais/kohya_ss
104 lines
3.4 KiB
Plaintext
104 lines
3.4 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {
|
|
"id": "view-in-github"
|
|
},
|
|
"source": [
|
|
"[](https://colab.research.google.com/github/bmaltais/kohya_ss/blob/feature/update-gui-uv-for-colab-parity/kohya_ss_colab.ipynb)"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"#@title Train with Kohya's Stable Diffusion Trainers\n",
|
|
"%cd /content\n",
|
|
"\n",
|
|
"import os\n",
|
|
"from google.colab import drive\n",
|
|
"drive.mount('/content/drive')\n",
|
|
"\n",
|
|
"%cd /content\n",
|
|
"!git clone -b dev https://github.com/bmaltais/kohya_ss\n",
|
|
"%cd /content/kohya_ss\n",
|
|
"!git checkout dev\n",
|
|
"\n",
|
|
"os.environ[\"TERM\"] = \"dumb\"\n",
|
|
"!./gui-uv.sh --quiet --share --headless"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"#@title Convert Safetensors to Diffusers\n",
|
|
"from_safetensors_url = '' #@param {type:\"string\"}\n",
|
|
"!wget -q https://raw.githubusercontent.com/huggingface/diffusers/v0.17.1/scripts/convert_original_stable_diffusion_to_diffusers.py\n",
|
|
"!wget {from_safetensors_url} -O /content/model.safetensors\n",
|
|
"!python3 convert_original_stable_diffusion_to_diffusers.py --half --from_safetensors --checkpoint_path model.safetensors --dump_path /content/model"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"#@title Push to HF.co\n",
|
|
"\n",
|
|
"import os\n",
|
|
"from huggingface_hub import create_repo, upload_folder\n",
|
|
"\n",
|
|
"hf_token = 'HF_WRITE_TOKEN' #@param {type:\"string\"}\n",
|
|
"repo_id = 'username/reponame' #@param {type:\"string\"}\n",
|
|
"commit_message = '\\u2764' #@param {type:\"string\"}\n",
|
|
"create_repo(repo_id, private=True, token=hf_token)\n",
|
|
"model_path = '/content/train/model' #@param {type:\"string\"}\n",
|
|
"upload_folder(folder_path=f'{model_path}', path_in_repo='', repo_id=repo_id, commit_message=commit_message, token=hf_token)"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"#@title Push to DagsHub.com\n",
|
|
"\n",
|
|
"!pip -q install dagshub\n",
|
|
"from dagshub.upload import Repo, create_repo\n",
|
|
"\n",
|
|
"repo_id = 'reponame' #@param {type:\"string\"}\n",
|
|
"org_name = 'orgname' #@param {type:\"string\"}\n",
|
|
"commit_message = '\\u2764' #@param {type:\"string\"}\n",
|
|
"create_repo(f\"{repo_id}\", org_name=f\"{org_name}\")\n",
|
|
"repo = Repo(f\"{org_name}\", f\"{repo_id}\")\n",
|
|
"model_path = '/content/train/model' #@param {type:\"string\"}\n",
|
|
"repo.upload(\"/content/models\", remote_path=\"data\", commit_message=f\"{commit_message}\", force=True)"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {
|
|
"accelerator": "GPU",
|
|
"colab": {
|
|
"gpuType": "T4",
|
|
"provenance": []
|
|
},
|
|
"kernelspec": {
|
|
"display_name": "Python 3",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"name": "python"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 0
|
|
}
|