Update README.md
Browse files
README.md
CHANGED
|
@@ -43,6 +43,9 @@ pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion
|
|
| 43 |
# Load the fine-tuned LoRA weights
|
| 44 |
pipe.load_lora_weights("Bhaskar009/SD_1.5_LoRA")
|
| 45 |
|
|
|
|
|
|
|
|
|
|
| 46 |
# Define a Naruto-themed prompt
|
| 47 |
prompt = "A detailed anime-style portrait of Naruto Uzumaki, wearing his Hokage cloak, standing under a bright sunset, ultra-detailed, cinematic lighting, 8K"
|
| 48 |
|
|
|
|
| 43 |
# Load the fine-tuned LoRA weights
|
| 44 |
pipe.load_lora_weights("Bhaskar009/SD_1.5_LoRA")
|
| 45 |
|
| 46 |
+
# moving to cuda
|
| 47 |
+
pipe.to("cuda")
|
| 48 |
+
|
| 49 |
# Define a Naruto-themed prompt
|
| 50 |
prompt = "A detailed anime-style portrait of Naruto Uzumaki, wearing his Hokage cloak, standing under a bright sunset, ultra-detailed, cinematic lighting, 8K"
|
| 51 |
|