Improve model card (#13)

- Improve model card (c366383ca277bd86209cc1b491de0ee2c9740561)


Co-authored-by: Patrick von Platen <[email protected]>

README.md

@@ -1,17 +1,58 @@
 ---
-license: creativeml-openrail-m
 language:
 - en
-library_name: diffusers
 tags:
+- stable-diffusion
+- stable-diffusion-diffusers
 - text-to-image
+- art
+- artistic
+- diffusers
+inference: true
+license: creativeml-openrail-m
 ---
-### ProtoGen Diffusion model merged by [darkstorm2150](https://instagram.com/officialvictorespinoza)
 
-This model was merged on a large amount of data from large datasets new and trending on civitai.com
+# Protogen_x2.2 by [darkstorm2150](https://instagram.com/officialvictorespinoza)
+
+Protogen was warm-started with [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) and fine-tuned on a large amount 
+of data from large datasets new and trending on civitai.com.
 
 You can enforce camera capture by using the prompt with "modelshoot style".
 
-It should also be very dreamboothable, being able to generate high fidelity faces with a little amount of steps.
+It should also be very "dreambooth-able", being able to generate high fidelity faces with a little amount of steps (see [dreambooth](https://github.com/huggingface/diffusers/tree/main/examples/dreambooth)).
+
+## Space
+
+We support a [Gradio](https://github.com/gradio-app/gradio) Web UI to run dreamlike-diffusion-1.0:
+[![Open In Spaces](https://camo.githubusercontent.com/00380c35e60d6b04be65d3d94a58332be5cc93779f630bcdfc18ab9a3a7d3388/68747470733a2f2f696d672e736869656c64732e696f2f62616467652f25463025394625413425393725323048756767696e67253230466163652d5370616365732d626c7565)](https://huggingface.co/spaces/darkstorm2150/Stable-Diffusion-Protogen-webui)
+### CompVis
+
+[Download Protogen_V2.2.ckpt) (5.98GB)](https://huggingface.co/darkstorm2150/Protogen_v2.2_Official_Release/blob/main/Protogen_V2.2.ckpt)
+
+### 🧨 Diffusers
+
+This model can be used just like any other Stable Diffusion model. For more information,
+please have a look at the [Stable Diffusion Pipeline](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
+
+```python
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
+import torch
+
+prompt = (
+"modelshoot style, (extremely detailed CG unity 8k wallpaper), full shot body photo of the most beautiful artwork in the world, "
+"english medieval witch, black silk vale, pale skin, black silk robe, black cat, necromancy magic, medieval era, "
+"photorealistic painting by Ed Blinkey, Atey Ghailan, Studio Ghibli, by Jeremy Mann, Greg Manchess, Antonio Moro, trending on ArtStation, "
+"trending on CGSociety, Intricate, High Detail, Sharp focus, dramatic, photorealistic painting art by midjourney and greg rutkowski"
+)
+
+model_id = "darkstorm2150/Protogen_v2.2_Official_Release"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+pipe = pipe.to("cuda")
+
+image = pipe(prompt, num_inference_steps=25).images[0]
+
+image.save("./result.jpg")
+```
 
 **By using this model you agree to this, That I the creator darkstorm2150 of this merge and Hugging Face is not liable for any content created by this Protogen Model.**