| from diffusers import DiffusionPipeline | |
| import torch.nn as nn, torch, types | |
| MODEL_DIR = "opendiffusionai/stablediffusion_t5" | |
| pipe = DiffusionPipeline.from_pretrained( | |
| MODEL_DIR, custom_pipeline=MODEL_DIR, use_safetensors=True, | |
| torch_dtype=torch.bfloat16, | |
| ) | |
| print("model initialized.") | |
| pipe.enable_sequential_cpu_offload() | |
| print("Trying render now...") | |
| images = pipe("a misty Tokyo alley at night",num_inference_steps=30).images | |
| fname="save.png" | |
| print(f"saving to {fname}") | |
| images[0].save(fname) | |