No official torrent yet β requesting it tells the flock what to host next. You cannot seed a catalog-only page.
Cataloged from zai-org/CogView4-6B
All models by zai-org β
π€ Space | π Github | π CogView3 Paper
512px and 2048px, divisible by 32, and ensure the maximum number of
pixels does not exceed 2^21 px.Using BF16 precision with batchsize=4 for testing, the memory usage is shown in the table below:
First, ensure you install the diffusers library from source.
Then, run the following code:
We've tested on multiple benchmarks and achieved the following scores:
π If you find our work helpful, please consider citing our paper and leaving valuable stars
This model is released under the Apache 2.0 License [blocked].
pip install git+https://github.com/huggingface/diffusers.git
cd diffusers
pip install -e .from diffusers import CogView4Pipeline
pipe = CogView4Pipeline.from_pretrained("THUDM/CogView4-6B", torch_dtype=torch.bfloat16)
# Open it for reduce GPU memory usage
pipe.enable_model_cpu_offload()
pipe.vae.enable_slicing()
pipe.vae.enable_tiling()
prompt = "A vibrant cherry red sports car sits proudly under the gleaming sun, its polished exterior smooth and flawless, casting a mirror-like reflection. The car features a low, aerodynamic body, angular headlights that gaze forward like predatory eyes, and a set of black, high-gloss racing rims that contrast starkly with the red. A subtle hint of chrome embellishes the grille and exhaust, while the tinted windows suggest a luxurious and private interior. The scene conveys a sense of speed and elegance, the car appearing as if it's about to burst into a sprint along a coastal road, with the ocean's azure waves crashing in the background."
image = pipe(
prompt=prompt,
guidance_scale=3.5,
num_images_per_prompt=1,
num_inference_steps=50,
width=1024,
height=1024,
).images[0]
image.save("cogview4.png")@article{zheng2024cogview3,
title={Cogview3: Finer and faster text-to-image generation via relay diffusion},
author={Zheng, Wendi and Teng, Jiayan and Yang, Zhuoyi and Wang, Weihan and Chen, Jidong and Gu, Xiaotao and Dong, Yuxiao and Ding, Ming and Tang, Jie},
journal={arXiv preprint arXiv:2403.05121},
year={2024}
}