Sebenr7 commited on
Commit
c36b9c3
1 Parent(s): 55fd50b

import torch from torch import autocast from diffusers import StableDiffusionPipeline pipe = StableDiffusionPipeline.from_pretrained( 'hakurei/waifu-diffusion', torch_dtype=torch.float32 ).to('cuda') prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt" with autocast("cuda"): image = pipe(prompt, guidance_scale=6)["sample"][0]

Browse files
Files changed (1) hide show
  1. README.md +0 -75
README.md DELETED
@@ -1,75 +0,0 @@
1
- ---
2
- language:
3
- - en
4
- tags:
5
- - stable-diffusion
6
- - text-to-image
7
- license: creativeml-openrail-m
8
- inference: true
9
-
10
- ---
11
-
12
- # waifu-diffusion v1.4 - Diffusion for Weebs
13
-
14
- waifu-diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning.
15
-
16
- ![image](https://user-images.githubusercontent.com/26317155/210155933-db3a5f1a-1ec3-4777-915c-6deff2841ce9.png)
17
-
18
- <sub>masterpiece, best quality, 1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck</sub>
19
-
20
- [Original Weights](https://huggingface.co/hakurei/waifu-diffusion-v1-4)
21
-
22
- # Gradio & Colab
23
-
24
- We also support a [Gradio](https://github.com/gradio-app/gradio) Web UI and Colab with Diffusers to run Waifu Diffusion:
25
- [![Open In Spaces](https://camo.githubusercontent.com/00380c35e60d6b04be65d3d94a58332be5cc93779f630bcdfc18ab9a3a7d3388/68747470733a2f2f696d672e736869656c64732e696f2f62616467652f25463025394625413425393725323048756767696e67253230466163652d5370616365732d626c7565)](https://huggingface.co/spaces/hakurei/waifu-diffusion-demo)
26
- [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1_8wPN7dJO746QXsFnB09Uq2VGgSRFuYE#scrollTo=1HaCauSq546O)
27
-
28
- ## Model Description
29
-
30
- [See here for a full model overview.](https://gist.github.com/harubaru/f727cedacae336d1f7877c4bbe2196e1)
31
-
32
- ## License
33
-
34
- This model is open access and available to all, with a CreativeML OpenRAIL-M license further specifying rights and usage.
35
- The CreativeML OpenRAIL License specifies:
36
-
37
- 1. You can't use the model to deliberately produce nor share illegal or harmful outputs or content
38
- 2. The authors claims no rights on the outputs you generate, you are free to use them and are accountable for their use which must not go against the provisions set in the license
39
- 3. You may re-distribute the weights and use the model commercially and/or as a service. If you do, please be aware you have to include the same use restrictions as the ones in the license and share a copy of the CreativeML OpenRAIL-M to all your users (please read the license entirely and carefully)
40
- [Please read the full license here](https://huggingface.co/spaces/CompVis/stable-diffusion-license)
41
-
42
- ## Downstream Uses
43
-
44
- This model can be used for entertainment purposes and as a generative art assistant.
45
-
46
- ## Example Code
47
-
48
- ```python
49
- import torch
50
- from torch import autocast
51
- from diffusers import StableDiffusionPipeline
52
-
53
- pipe = StableDiffusionPipeline.from_pretrained(
54
- 'hakurei/waifu-diffusion',
55
- torch_dtype=torch.float32
56
- ).to('cuda')
57
-
58
- prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
59
- with autocast("cuda"):
60
- image = pipe(prompt, guidance_scale=6)["sample"][0]
61
-
62
- image.save("test.png")
63
- ```
64
-
65
- ## Team Members and Acknowledgements
66
-
67
- This project would not have been possible without the incredible work by Stability AI and Novel AI.
68
-
69
- - [Haru](https://github.com/harubaru)
70
- - [Salt](https://github.com/sALTaccount/)
71
- - [Sta @ Bit192](https://twitter.com/naclbbr)
72
-
73
- In order to reach us, you can join our [Discord server](https://discord.gg/touhouai).
74
-
75
- [![Discord Server](https://discordapp.com/api/guilds/930499730843250783/widget.png?style=banner2)](https://discord.gg/touhouai)