|
--- |
|
tags: |
|
- dreambooth-hackathon |
|
--- |
|
中国奇谭第一季画风模型 |
|
|
|
import torch |
|
|
|
from diffusers import StableDiffusionPipeline |
|
|
|
pipe = StableDiffusionPipeline.from_pretrained("sd-dreambooth-library/cnstory", torch_dtype=torch.float16, use_auth_token=True) |
|
|
|
pipe = pipe.to("cuda") |
|
|
|
prompt = "a photograph of an astronaut riding a horse, cnstory artstyle" |
|
|
|
image = pipe(prompt).images[0] |
|
|
|
image.save(f"astronaut_rides_horse.png") |
|
|
|
 |
|
 |
|
 |
|
 |
