Views
No views yet
1pip install diffusers transformers accelerate imageio[ffmpeg]
2pip install git+ssh://git@github.com/baaivision/URSA.git1import torch
2from diffnext.pipelines import URSAPipeline
3
4model_id, height, width = "BAAI/URSA-1.7B-IBQ1024", 1024, 1024
5model_args = {"torch_dtype": torch.float16, "trust_remote_code": True}
6pipe = URSAPipeline.from_pretrained(model_id, **model_args)
7pipe = pipe.to(torch.device("cuda"))
8
9prompt = "The bear, calm and still, gazes upward as if lost in contemplation of the cosmos."
10negative_prompt = "worst quality, low quality, inconsistent motion, static, still, blurry, jittery, distorted, ugly"
11
12image = pipe(**locals()).frames[0]
13image.save("ursa.jpg")