Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ It was finetuned in order to create after a pipeline of prompt-to-3D model.
|
|
| 32 |
#### How to use
|
| 33 |
|
| 34 |
```python
|
| 35 |
-
#
|
| 36 |
import torch
|
| 37 |
import os
|
| 38 |
from diffusers import StableDiffusionPipeline
|
|
@@ -43,7 +43,6 @@ pipe.unet.load_attn_procs('remi349/sd_trained_3D_lora')
|
|
| 43 |
pipe.to("cuda")
|
| 44 |
|
| 45 |
# Then you can begin the inference process on a prompt and save the image generated
|
| 46 |
-
|
| 47 |
prompt = 'a rabbit with a yellow jacket'
|
| 48 |
image = pipe(prompt, num_inference_steps=30, guidance_scale=7.5).images[0]
|
| 49 |
image.save("my_image.png")
|
|
|
|
| 32 |
#### How to use
|
| 33 |
|
| 34 |
```python
|
| 35 |
+
# First load the basic architecture and everything
|
| 36 |
import torch
|
| 37 |
import os
|
| 38 |
from diffusers import StableDiffusionPipeline
|
|
|
|
| 43 |
pipe.to("cuda")
|
| 44 |
|
| 45 |
# Then you can begin the inference process on a prompt and save the image generated
|
|
|
|
| 46 |
prompt = 'a rabbit with a yellow jacket'
|
| 47 |
image = pipe(prompt, num_inference_steps=30, guidance_scale=7.5).images[0]
|
| 48 |
image.save("my_image.png")
|