| from deepfloyd_if.modules import IFStageI, IFStageII, StableStageIII |
| from deepfloyd_if.modules.t5 import T5Embedder |
| from deepfloyd_if.pipelines import dream |
|
|
| |
| device = 'cuda' |
| cache_dir = "/path/to/storage/IF" |
| if_I = IFStageI('IF-I-L-v1.0', device=device, cache_dir=cache_dir) |
| if_II = IFStageII('IF-II-L-v1.0', device=device, cache_dir=cache_dir) |
| if_III = StableStageIII('stable-diffusion-x4-upscaler', device=device, cache_dir=cache_dir) |
| t5 = T5Embedder(device=device, cache_dir=cache_dir) |
|
|
| prompt = "In the heart of the wilderness, an enchanting forest reveals itself. \ |
| Towering trees, their trunks sturdy and thick, reach skyward, their leafy canopies \ |
| forming a natural cathedral. Verdant moss clings to bark, and tendrils of ivy climb ambitiously towards the sun-dappled treetops. \ |
| The forest floor is a tapestry of fallen leaves, sprinkled with delicate wildflowers. The soft chatter of wildlife resonates, while a nearby brook babbles, its clear waters winking in the dappled light. \ |
| Sunrays filter through the foliage, casting an emerald glow that dances on the woodland floor. Amidst the tranquility, the forest teems with life, whispering ancient secrets on the breeze." |
| count = 1 |
|
|
| result = dream( |
| t5=t5, if_I=if_I, if_II=if_II, if_III=if_III, |
| prompt=[prompt]*count, |
| seed=42, |
| if_I_kwargs={ |
| "guidance_scale": 7.0, |
| "sample_timestep_respacing": "smart100", |
| }, |
| if_II_kwargs={ |
| "guidance_scale": 4.0, |
| "sample_timestep_respacing": "smart50", |
| }, |
| if_III_kwargs={ |
| "guidance_scale": 9.0, |
| "noise_level": 20, |
| "sample_timestep_respacing": "75", |
| }, |
| ) |
| result['III'][0].save("./dream_figure.jpg") |
|
|
|
|
