Commit
•
2e68b80
1
Parent(s):
52751ea
Update README.md
Browse files
README.md
CHANGED
|
@@ -29,7 +29,7 @@ from diffusers import StableDiffusionPipeline, TCDScheduler
|
|
| 29 |
pipeline = StableDiffusionPipeline.from_pretrained("Lykon/dreamshaper-7", torch_dtype=torch.float16, safety_checker = None).to("cuda")
|
| 30 |
pipeline.load_lora_weights("rootonchair/tscd-dreamshaper-7")
|
| 31 |
pipeline.scheduler = TCDScheduler.from_config(pipeline.scheduler.config)
|
| 32 |
-
test_prompts
|
| 33 |
"cgmech, white mecha robot, cape, science fiction, torn clothes, glowing, standing, robot joints, mecha, armor, cowboy shot, intense sunlight, silver dragonborn, outdoors, landscape, nature, volumetrics dtx",
|
| 34 |
"Portrait photo of muscular bearded guy in a worn mech suit, elegant, sharp focus, photo by greg rutkowski, soft lighting, vibrant colors",
|
| 35 |
"photo of a supercar, 8k uhd, high quality, road, sunset, motion blur, depth blur, cinematic, filmic image 4k",
|
|
@@ -39,7 +39,7 @@ test_prompts = [
|
|
| 39 |
"Self-portrait oil painting, a beautiful cyborg with golden hair, 8k",
|
| 40 |
"Astronaut in a jungle, cold color palette, muted colors, detailed, 8k";
|
| 41 |
]
|
| 42 |
-
seed
|
| 43 |
images = pipeline(test_prompts, width=512, height=512, num_inference_steps=4, guidance_scale=0, generator=torch.Generator(device="cuda").manual_seed(seed)).images
|
| 44 |
```
|
| 45 |
|
|
|
|
| 29 |
pipeline = StableDiffusionPipeline.from_pretrained("Lykon/dreamshaper-7", torch_dtype=torch.float16, safety_checker = None).to("cuda")
|
| 30 |
pipeline.load_lora_weights("rootonchair/tscd-dreamshaper-7")
|
| 31 |
pipeline.scheduler = TCDScheduler.from_config(pipeline.scheduler.config)
|
| 32 |
+
test_prompts = [
|
| 33 |
"cgmech, white mecha robot, cape, science fiction, torn clothes, glowing, standing, robot joints, mecha, armor, cowboy shot, intense sunlight, silver dragonborn, outdoors, landscape, nature, volumetrics dtx",
|
| 34 |
"Portrait photo of muscular bearded guy in a worn mech suit, elegant, sharp focus, photo by greg rutkowski, soft lighting, vibrant colors",
|
| 35 |
"photo of a supercar, 8k uhd, high quality, road, sunset, motion blur, depth blur, cinematic, filmic image 4k",
|
|
|
|
| 39 |
"Self-portrait oil painting, a beautiful cyborg with golden hair, 8k",
|
| 40 |
"Astronaut in a jungle, cold color palette, muted colors, detailed, 8k";
|
| 41 |
]
|
| 42 |
+
seed = 1234
|
| 43 |
images = pipeline(test_prompts, width=512, height=512, num_inference_steps=4, guidance_scale=0, generator=torch.Generator(device="cuda").manual_seed(seed)).images
|
| 44 |
```
|
| 45 |
|