zai-org
/

CogView3-Plus-3B

CogView3PlusPipeline

image-generation

Model card Files Files and versions

hlky commited on Dec 22, 2024

Commit

565772a

·

verified ·

1 Parent(s): 5d70e40

Use `bfloat16` in the example

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -62,7 +62,7 @@ Then, run the following code:
 from diffusers import CogView3PlusPipeline
 import torch
-pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3-Plus-3B", torch_dtype=torch.float16).to("cuda")
 # Enable it to reduce GPU memory usage
 pipe.enable_model_cpu_offload()

 from diffusers import CogView3PlusPipeline
 import torch
+pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3-Plus-3B", torch_dtype=torch.bfloat16).to("cuda")
 # Enable it to reduce GPU memory usage
 pipe.enable_model_cpu_offload()