Update README.md
Browse files
README.md
CHANGED
@@ -67,8 +67,15 @@ You should use `an icon of trpfrog` to trigger the image generation.
|
|
67 |
```py
|
68 |
from diffusers import AutoPipelineForText2Image
|
69 |
import torch
|
70 |
-
|
71 |
-
pipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
image = pipeline('an icon of trpfrog eating ramen').images[0]
|
73 |
```
|
74 |
|
|
|
67 |
```py
|
68 |
from diffusers import AutoPipelineForText2Image
|
69 |
import torch
|
70 |
+
|
71 |
+
pipeline = AutoPipelineForText2Image.from_pretrained(
|
72 |
+
stabilityai/stable-diffusion-3.5-medium,
|
73 |
+
torch_dtype=torch.float16
|
74 |
+
).to('cuda')
|
75 |
+
pipeline.load_lora_weights(
|
76 |
+
'Prgckwb/trpfrog-sd3.5m-lora',
|
77 |
+
weight_name='pytorch_lora_weights.safetensors'
|
78 |
+
)
|
79 |
image = pipeline('an icon of trpfrog eating ramen').images[0]
|
80 |
```
|
81 |
|