Using the
🤗's Diffusers library to run NOVA in a simple and efficient manner.
1pip install diffusers transformers accelerate
2pip install git+ssh://git@github.com/baaivision/NOVA.git
1import torch
2from diffnext.pipelines import NOVAPipeline
3
4model_id = "BAAI/nova-d48w1536-sdxl1024"
5model_args = {"torch_dtype": torch.float16, "trust_remote_code": True}
6pipe = NOVAPipeline.from_pretrained(model_id, **model_args)
7pipe = pipe.to("cuda")
8
9prompt = "a shiba inu wearing a beret and black turtleneck."
10image = pipe(prompt).images[0]
11
12image.save("shiba_inu.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.