Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -53,7 +53,9 @@ Weights for this model are available in Safetensors format.
|
|
| 53 |
from diffusers import AutoPipelineForText2Image
|
| 54 |
import torch
|
| 55 |
|
| 56 |
-
|
|
|
|
|
|
|
| 57 |
pipeline.load_lora_weights('dfrer/nanasa', weight_name='nanasa-000009.safetensors')
|
| 58 |
image = pipeline('Lady death').images[0]
|
| 59 |
```
|
|
|
|
| 53 |
from diffusers import AutoPipelineForText2Image
|
| 54 |
import torch
|
| 55 |
|
| 56 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 57 |
+
|
| 58 |
+
pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-2-1-base', torch_dtype=torch.float16).to(device)
|
| 59 |
pipeline.load_lora_weights('dfrer/nanasa', weight_name='nanasa-000009.safetensors')
|
| 60 |
image = pipeline('Lady death').images[0]
|
| 61 |
```
|