import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("neanderthalai/my-awesome-model", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Model Name
Brief description of what your model does.
Model Details
Model Description
- Developed by: Your Name/Organization
- Model type: Diffusion model
- Language(s): English
- License: Apache 2.0
- Finetuned from model: stabilityai/stable-diffusion-2-1
Model Sources
- Repository: https://github.com/your-repo
- Paper: Link to paper if applicable
- Demo: Link to demo if available
Uses
Direct Use
Describe how the model can be used directly.
Downstream Use
Describe potential downstream applications.
Out-of-Scope Use
Describe uses that are out of scope or misuse of the model.
Bias, Risks, and Limitations
Document known biases, risks, and limitations.
How to Get Started with the Model
from diffusers import DDPMPipeline
pipeline = DDPMPipeline.from_pretrained({hub_model_id})
image = pipeline().images[0]
image
- Downloads last month
- -