Update README.md
Browse files
README.md
CHANGED
@@ -75,7 +75,7 @@ from diffusers import DiffusionPipeline
|
|
75 |
import torch
|
76 |
|
77 |
base_model_id = 'stabilityai/stable-diffusion-xl-base-1.0'
|
78 |
-
lora_model_id = 'Prgckwb/trpfrog-
|
79 |
|
80 |
pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16)
|
81 |
pipe = pipe.to("cuda")
|
|
|
75 |
import torch
|
76 |
|
77 |
base_model_id = 'stabilityai/stable-diffusion-xl-base-1.0'
|
78 |
+
lora_model_id = 'Prgckwb/trpfrog-sdxl-lora'
|
79 |
|
80 |
pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16)
|
81 |
pipe = pipe.to("cuda")
|