diff --git "a/test.ipynb" "b/test.ipynb" --- "a/test.ipynb" +++ "b/test.ipynb" @@ -6,20 +6,10 @@ "id": "242d5fad-d25c-4ed0-94d3-9daa49fe44ed", "metadata": {}, "outputs": [ - { - "name": "stdout", - "output_type": "stream", - "text": [ - "Total memory: 15.70 GB\n", - "Free memory: 15.46 GB\n" - ] - }, { "name": "stderr", "output_type": "stream", "text": [ - "Some weights of XLMRobertaModel were not initialized from the model checkpoint at /home/recoilme/models/waifu-2b and are newly initialized: ['pooler.dense.bias', 'pooler.dense.weight']\n", - "You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.\n", "/home/recoilme/.local/lib/python3.11/site-packages/xformers/ops/fmha/flash.py:211: FutureWarning: `torch.library.impl_abstract` was renamed to `torch.library.register_fake`. Please use that instead; we will remove `torch.library.impl_abstract` in a future version of PyTorch.\n", " @torch.library.impl_abstract(\"xformers_flash::flash_fwd\")\n", "/home/recoilme/.local/lib/python3.11/site-packages/xformers/ops/fmha/flash.py:344: FutureWarning: `torch.library.impl_abstract` was renamed to `torch.library.register_fake`. Please use that instead; we will remove `torch.library.impl_abstract` in a future version of PyTorch.\n", @@ -29,12 +19,12 @@ { "data": { "application/vnd.jupyter.widget-view+json": { - "model_id": "fa52e90425504765923a553d70494fa7", + "model_id": "286393babfbc4673a72526b00c8cd8af", "version_major": 2, "version_minor": 0 }, "text/plain": [ - "Loading pipeline components...: 0%| | 0/3 [00:00" + "" ] }, "metadata": {}, @@ -183,35 +148,12 @@ } ], "source": [ - "prompt = '1girl, solo, animal ears, bow, teeth, jacket, tail, open mouth, brown hair, orange background, bowtie, orange nails, simple background, cat ears, orange eyes, blue bow, animal ear fluff, cat tail, looking at viewer, upper body, shirt, school uniform, hood, striped bow, striped, white shirt, black jacket, blue bowtie, fingernails, long sleeves, cat girl, bangs, fangs, collared shirt, striped bowtie, short hair, tongue, hoodie, sharp teeth, facial mark, claw pose'\n", - "prompt = 'A young girl with long brown hair and brown eyes is standing against a brick wall, wearing a white shirt with a black sailor collar and a black pleated skirt. She is smiling and looking directly at the viewer, with sunlight filtering through the green vines behind her'\n", - "#prompt = 'нарядная новогодняя елка, красивые игрушки, звезда сверху, огоньки, на тёмном фоне' \n", - "prompt_embeds, prompt_attention_mask = txt_embeds(prompt)\n", - "negative_prompt_embeds, negative_prompt_attention_mask=txt_embeds(\"\")\n", - "print(prompt_embeds.shape)\n", - "#del text_encoder\n", - "#del tokenizer\n", - "import gc\n", - "gc.collect() # Принудительно вызываем сборщик мусора\n", - "torch.cuda.empty_cache()\n", - "import torch\n", - "\n", - "free_memory, total_memory = torch.cuda.mem_get_info()\n", - "print(f\"Total memory: {total_memory / 1024 ** 3:.2f} GB\")\n", - "print(f\"Free memory: {free_memory / 1024 ** 3:.2f} GB\")\n", + "prompt = 'аниме девушка, waifu, يبتسم جنسيا , sur le fond de la tour Eiffel'\n", + "generator = torch.Generator(device=\"cuda\").manual_seed(42)\n", "\n", "image = pipeline(\n", - " prompt = None,\n", - " negative_prompt = None,\n", - " use_resolution_binning = False,\n", - " prompt_embeds=prompt_embeds,\n", - " prompt_attention_mask= prompt_attention_mask,\n", - " negative_prompt_embeds = negative_prompt_embeds,\n", - " negative_prompt_attention_mask = negative_prompt_attention_mask,\n", - " height=768,\n", - " width=768,\n", - " guidance_scale=4.5,\n", - " num_inference_steps=24,\n", + " prompt = prompt,\n", + " negative_prompt = \"\",\n", " generator=torch.Generator(device=\"cuda\").manual_seed(42),\n", ")[0]\n", "\n", @@ -222,7 +164,7 @@ { "cell_type": "code", "execution_count": null, - "id": "13392411-a64a-4a23-9a2a-3ec0bb7629c9", + "id": "c3f41f60-eb16-4132-9db7-a366c3fe7007", "metadata": {}, "outputs": [], "source": []