Downloads · 30 days
11
1% of all-time downloads
BAAI/URSA-0.6B-IBQ1024
URSA-0.6B-IBQ1024 is a text-to-image model from BAAI. Use it when you need an image from a text prompt. It is set up for diffusers. The card lists the license as apache-2.0.
- Developed by: BAAI - Model type: Text-to-Image Generation Model - Model size: 0.6B - Model precision: torch.float16 (FP16) - Model resolution: 1024x1024 - Model paper: Uniform Discrete Diffusion with Metric Path for…
Downloads · 30 days
11
1% of all-time downloads
All-time downloads
973
Public
Parameters
864M
2.8 GB on disk
Likes
6
Public
Click a slice to open those files.
.safetensors2.6 GB · 95%
From the Hugging Face model README
Using the 🤗's Diffusers library to run URSA in a simple and efficient manner.
pip install diffusers transformers accelerate imageio[ffmpeg]
pip install git+ssh://[email protected]/baaivision/URSA.git
Running the pipeline:
import torch
from diffnext.pipelines import URSAPipeline
model_id, height, width = "BAAI/URSA-0.6B-IBQ1024", 1024, 1024
model_args = {"torch_dtype": torch.float16, "trust_remote_code": True}
pipe = URSAPipeline.from_pretrained(model_id, **model_args)
pipe = pipe.to(torch.device("cuda"))
prompt = "The bear, calm and still, gazes upward as if lost in contemplation of the cosmos."
negative_prompt = "worst quality, low quality, inconsistent motion, static, still, blurry, jittery, distorted, ugly"
image = pipe(**locals()).frames[0]
image.save("ursa.jpg")
The model is intended for research purposes only. Possible research areas and tasks include
Excluded uses are described below.
The model was not trained to be factual or true representations of people or events, and therefore using the model to generate such content is out-of-scope for the abilities of this model.
Using the model to generate content that is cruel to individuals is a misuse of this model. This includes, but is not limited to:
While the capabilities of image generation models are impressive, they can also reinforce or exacerbate social biases.