{ "cells": [ { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "from diffusers import AutoPipelineForText2Image\n", "import torch" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "vae/diffusion_pytorch_model.safetensors not found\n" ] }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "1822a5456c3244b6b5831817d6d0ebbc", "version_major": 2, "version_minor": 0 }, "text/plain": [ "Fetching 15 files: 0%| | 0/15 [00:00<?, ?it/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "6232a45e054e4439a822345b4744b0b8", "version_major": 2, "version_minor": 0 }, "text/plain": [ "tokenizer/special_tokens_map.json: 0%| | 0.00/472 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "da77db92e94e482b83b679b698630be0", "version_major": 2, "version_minor": 0 }, "text/plain": [ "(…)ature_extractor/preprocessor_config.json: 0%| | 0.00/342 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "1f73d5e96b8b4c218433d8c9ea66b4b1", "version_major": 2, "version_minor": 0 }, "text/plain": [ "safety_checker/config.json: 0%| | 0.00/4.80k [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "4eb1d6d2c98c46e6940129236928b9d8", "version_major": 2, "version_minor": 0 }, "text/plain": [ "text_encoder/config.json: 0%| | 0.00/589 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "8455fd4bd49e462e94a15614968edeeb", "version_major": 2, "version_minor": 0 }, "text/plain": [ "tokenizer/merges.txt: 0%| | 0.00/525k [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "85c9b8ddc3084051b9de01a1cc4b346f", "version_major": 2, "version_minor": 0 }, "text/plain": [ "scheduler/scheduler_config.json: 0%| | 0.00/284 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "f01d373de73e49198ed9a509ff884d83", "version_major": 2, "version_minor": 0 }, "text/plain": [ "unet/config.json: 0%| | 0.00/789 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "2dad6d09e0b944cb91e5e68d374ad283", "version_major": 2, "version_minor": 0 }, "text/plain": [ "tokenizer/vocab.json: 0%| | 0.00/1.06M [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "69c41904a7dd439296a3a56c0a37acda", "version_major": 2, "version_minor": 0 }, "text/plain": [ "vae/config.json: 0%| | 0.00/592 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "a3c8d76377f9445e81d73336d2b70535", "version_major": 2, "version_minor": 0 }, "text/plain": [ "safety_checker/pytorch_model.bin: 0%| | 0.00/608M [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "67a6fd052c734c69950ce38c7503821a", "version_major": 2, "version_minor": 0 }, "text/plain": [ "text_encoder/pytorch_model.bin: 0%| | 0.00/246M [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "e7de26e3010f484bb698f62dfae0a255", "version_major": 2, "version_minor": 0 }, "text/plain": [ "tokenizer/tokenizer_config.json: 0%| | 0.00/805 [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "d4080d04c64642a09514fd7570ba0cde", "version_major": 2, "version_minor": 0 }, "text/plain": [ "unet/diffusion_pytorch_model.bin: 0%| | 0.00/1.72G [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "402fd6d5c4054bdfae85f085c538ebac", "version_major": 2, "version_minor": 0 }, "text/plain": [ "vae/diffusion_pytorch_model.bin: 0%| | 0.00/167M [00:00<?, ?B/s]" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "# model = \"CompVis/ldm-text2im-large-256\"\n", "model = \"sd-dreambooth-library/colorful-ball\"\n", "# model = \"stabilityai/sdxl-turbo\"\n", "\n", "pipeline_text2image = AutoPipelineForText2Image.from_pretrained(\n", " model,\n", " torch_dtype=torch.float16,\n", ")\n", "pipeline_text2image = pipeline_text2image.to(\"cuda\")" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "gradio", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.12.2" } }, "nbformat": 4, "nbformat_minor": 2 }