NucleusAI
/

Nucleus-Image

NucleusMoEImagePipeline

Mixture of Experts

image-generation

Model card Files Files and versions

nucleus-ai commited on Apr 14

Commit

aae9948

·

verified ·

1 Parent(s): 648489e

Model name change

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -82,7 +82,7 @@ import torch
 from diffusers import DiffusionPipeline
 from diffusers import TextKVCacheConfig
-model_name = "NucleusAI/NucleusMoE-Image"
 pipe = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.bfloat16)
 pipe.to("cuda")

 from diffusers import DiffusionPipeline
 from diffusers import TextKVCacheConfig
+model_name = "NucleusAI/Nucleus-Image"
 pipe = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.bfloat16)
 pipe.to("cuda")