Update README.md
Browse files
README.md
CHANGED
@@ -34,10 +34,10 @@ These are LoRA adaption weights for CompVis/stable-diffusion-v1-4. The weights w
|
|
34 |
|
35 |
```python
|
36 |
pipe = DiffusionPipeline.from_pretrained(
|
37 |
-
|
38 |
)
|
39 |
|
40 |
-
pipe.load_lora_weights(
|
41 |
pipe.to('cuda')
|
42 |
|
43 |
prompt = "A histopathology image of breast cancer tissue"
|
|
|
34 |
|
35 |
```python
|
36 |
pipe = DiffusionPipeline.from_pretrained(
|
37 |
+
"CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16
|
38 |
)
|
39 |
|
40 |
+
pipe.load_lora_weights("RiddleHe/SD14_pathology_lora")
|
41 |
pipe.to('cuda')
|
42 |
|
43 |
prompt = "A histopathology image of breast cancer tissue"
|