Instructions to use openai/shap-e with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use openai/shap-e with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("openai/shap-e", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Commit ·
17c7a73
1
Parent(s): 1e72ed4
should save images[0] to gif file
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ images = pipe(
|
|
| 58 |
frame_size=256,
|
| 59 |
).images
|
| 60 |
|
| 61 |
-
gif_path = export_to_gif(images, "shark_3d.gif")
|
| 62 |
```
|
| 63 |
|
| 64 |
## Results
|
|
|
|
| 58 |
frame_size=256,
|
| 59 |
).images
|
| 60 |
|
| 61 |
+
gif_path = export_to_gif(images[0], "shark_3d.gif")
|
| 62 |
```
|
| 63 |
|
| 64 |
## Results
|