sippycoder commited on
Commit
1f31f70
·
1 Parent(s): 657b990

Update README.md (#1)

Browse files

- Update README.md (49322a2b8eff708a0ff6d51aa8c926f886f672ce)

Files changed (1) hide show
  1. README.md +33 -3
README.md CHANGED
@@ -1,3 +1,33 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: text-to-image
4
+ library_name: diffusers
5
+ ---
6
+
7
+ ![NucleusMoE-Image](demo.png)
8
+
9
+ NucleusMoE-Image is a 17B flow-based text-to-image generation MoE model with only 2B active parameters. Technical report coming soon...
10
+
11
+ ## Usage
12
+
13
+ ```python
14
+ import torch
15
+ from diffusers import DiffusionPipeline
16
+
17
+ pipe = DiffusionPipeline.from_pretrained("NucleusAI/NucleusMoE-Image", torch_dtype=torch.bfloat16)
18
+ pipe.to("cuda")
19
+
20
+ config = TextKVCacheConfig()
21
+ pipe.transformer.enable_cache(config)
22
+
23
+ prompt = "Vintage-style poster for Artificial Analysis depicting a retro-futuristic cityscape dominated by AI. Towering structures shaped like neural networks loom over sleek flying cars. In the foreground, a stylized robot extends a hand towards the viewer. Bold, sans-serif typography at the top reads 'Welcome to the AI Revolution' with 'Artificial Analysis' prominently displayed at the bottom in a retro chrome effect. The color palette consists of deep blues, vibrant oranges, and metallic silvers, reminiscent of 1950s sci-fi illustrations."
24
+ image = pipe(
25
+ prompt,
26
+ height=1024,
27
+ width=1024,
28
+ guidance_scale=4.0,
29
+ num_inference_steps=50,
30
+ ).images[0]
31
+ image.save("nucleus_image_demo.png")
32
+ ```
33
+