Commit
•
a883320
1
Parent(s):
121c639
Update README.md
Browse files
README.md
CHANGED
|
@@ -20,8 +20,7 @@ Version 3.4 continued training from [ProtoGen v2.2](https://huggingface.co/darks
|
|
| 20 |
## Space
|
| 21 |
|
| 22 |
We support a [Gradio](https://github.com/gradio-app/gradio) Web UI to run dreamlike-diffusion-1.0:
|
| 23 |
-
[](https://huggingface.co/spaces/
|
| 24 |
-
|
| 25 |
### CompVis
|
| 26 |
|
| 27 |
[Download ProtoGen_X3.4.ckpt) (5.98GB)](https://huggingface.co/darkstorm2150/Protogen_x3.4_Official_Release/blob/main/ProtoGen_X3.4.ckpt)
|
|
@@ -32,7 +31,7 @@ This model can be used just like any other Stable Diffusion model. For more info
|
|
| 32 |
please have a look at the [Stable Diffusion Pipeline](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
|
| 33 |
|
| 34 |
```python
|
| 35 |
-
from diffusers import StableDiffusionPipeline
|
| 36 |
import torch
|
| 37 |
|
| 38 |
prompt = (
|
|
@@ -44,9 +43,10 @@ prompt = (
|
|
| 44 |
|
| 45 |
model_id = "darkstorm2150/Protogen_x3.4_Official_Release"
|
| 46 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
|
|
|
| 47 |
pipe = pipe.to("cuda")
|
| 48 |
|
| 49 |
-
image = pipe(prompt).images[0]
|
| 50 |
|
| 51 |
image.save("./result.jpg")
|
| 52 |
```
|
|
|
|
| 20 |
## Space
|
| 21 |
|
| 22 |
We support a [Gradio](https://github.com/gradio-app/gradio) Web UI to run dreamlike-diffusion-1.0:
|
| 23 |
+
[](https://huggingface.co/spaces/darkstorm2150/Stable-Diffusion-Protogen-webui)
|
|
|
|
| 24 |
### CompVis
|
| 25 |
|
| 26 |
[Download ProtoGen_X3.4.ckpt) (5.98GB)](https://huggingface.co/darkstorm2150/Protogen_x3.4_Official_Release/blob/main/ProtoGen_X3.4.ckpt)
|
|
|
|
| 31 |
please have a look at the [Stable Diffusion Pipeline](https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion).
|
| 32 |
|
| 33 |
```python
|
| 34 |
+
from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
| 35 |
import torch
|
| 36 |
|
| 37 |
prompt = (
|
|
|
|
| 43 |
|
| 44 |
model_id = "darkstorm2150/Protogen_x3.4_Official_Release"
|
| 45 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
| 46 |
+
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 47 |
pipe = pipe.to("cuda")
|
| 48 |
|
| 49 |
+
image = pipe(prompt, num_inference_steps=25).images[0]
|
| 50 |
|
| 51 |
image.save("./result.jpg")
|
| 52 |
```
|