dalle2-pytorch

Files (27)

• SKILL.md
• repo-provenance.md · references/
• repo-routing-metadata.json · references/
• troubleshooting.md · references/
• check_install.py · scripts/
• SKILL.md · sub-skills/data-and-tracking/
• data-formats.md · sub-skills/data-and-tracking/references/
• tracking-and-checkpoints.md · sub-skills/data-and-tracking/references/
• troubleshooting.md · sub-skills/data-and-tracking/references/
• validate_webdataset_layout.py · sub-skills/data-and-tracking/scripts/
• SKILL.md · sub-skills/generation-and-api/
• api-reference.md · sub-skills/generation-and-api/references/
• latent-diffusion-and-vqgan.md · sub-skills/generation-and-api/references/
• troubleshooting.md · sub-skills/generation-and-api/references/
• workflows.md · sub-skills/generation-and-api/references/
• check_dalle2_runtime.py · sub-skills/generation-and-api/scripts/
• SKILL.md · sub-skills/training-and-configs/
• cli-reference.md · sub-skills/training-and-configs/references/
• decoder-cpu-smoke.json · sub-skills/training-and-configs/references/config-templates/
• prior-minimal.json · sub-skills/training-and-configs/references/config-templates/
• trainer-api.md · sub-skills/training-and-configs/references/
• training-configs.md · sub-skills/training-and-configs/references/
• troubleshooting.md · sub-skills/training-and-configs/references/
• inspect_training_config.py · sub-skills/training-and-configs/scripts/
• run_decoder_training.py · sub-skills/training-and-configs/scripts/
• run_diffusion_prior_training.py · sub-skills/training-and-configs/scripts/
• training_command_builder.py · sub-skills/training-and-configs/scripts/

SKILL.md (first lines)

# DALLE2-pytorch

Use this repo skill when a task involves the `dalle2-pytorch` package, DALL-E 2 style CLIP-latent text-to-image generation, diffusion prior training, cascaded decoder training, DALLE2 checkpoints, WebDataset embedding data, or the package's training/experiment tracking utilities.

## What This Skill Covers

- Public Python APIs for `DALLE2`, `DiffusionPriorNetwork`, `DiffusionPrior`, `Unet`, `Decoder`, CLIP adapters, VQGAN/VAE latent diffusion, inpainting, tokenization, and the `dream` CLI.
- JSON config classes and bundled training launchers for decoder and diffusion-prior training.
- WebDataset and EmbeddingReader data layouts, sidecar embedding folders, shard/index validation, trackers, checkpoint loading, and checkpoint saving.
- Troubleshooting for install/import, CLIP downloads, CPU vs GPU expectations, config validation, data layout, tracker credentials, and checkpoint format mismatches.

## Install And Minimal Check

Install the public package:

```bash
python -m pip install dalle2-pytorch
```

Run the bundled package check from this skill directory:

```bash
python scripts/check_install.py --mode imports
python scripts/check_install.py --mode cli-help
```

For a CPU-only synthetic API smoke test that avoids CLIP downloads and model weights:

```bash
python sub-skills/generation-and-api/scripts/check_dalle2_runtime.py --mode tiny-forward
```

## Choose A Sub-Skill

### `generation-and-api`

Read [sub-skills/generation-and-api/SKILL.md](sub-skills/generation-and-api/SKILL.md) for Python model construction and generation tasks:

- Build or call `DALLE2`, `DiffusionPrior`, `Decoder`, `Unet`, CLIP adapters, or `VQGanVAE`.
- Use `dream`, load generation checkpoints, tokenize prompts, sample images, or perform decoder inpainting.
- Debug checkpoint architecture mismatches, CLIP weight downloads, CPU/GPU limitations, inpaint mask shape, or classifier-free guidance.

### `training-and-configs`

Read [sub-skills/training-and-configs/SKILL.md](sub-skills/training-and-configs/SKILL.md) for training and configuration tasks:

- Author/validate `TrainDecoderConfig` or `TrainDiffusionPriorConfig` JSON.
- Use `DecoderTrainer` or `DiffusionPriorTrainer` directly.
- Build `python` or `accelerate launch` commands, run bundled training wrappers, resume/save trainer checkpoints, or debug `unet_training_mask`, resampling, DeepSpeed fp16, and config errors.

### `data-and-tracking`

Read [sub-skills/data-and-tracking/SKILL.md](sub-skills/data-and-tracking/SKILL.md) for data and experiment-management tasks:

- Validate decoder WebDataset shard names, sample keys, sidecar image/text embedding folders, `shard_width`, or `index_width`.
- Use `ImageEmbeddingDataset`, `create_image_embedding_dataloader`, `get_reader`, `make_splits`, or `PriorEmbeddingDataset`.
- Configure console/W&B logging, local/URL/W&B checkpoint loading, or local/W&B/HuggingFace saving.

## Important Constraints

- Real training and useful image sampling normally need GPU memory and trained checkpoints. CPU is suitable for imports, config parsing, CLI help, and tiny synthetic forward-loss tests.
- CLIP adapters (`OpenAIClipAdapter`, `OpenClipAdapter`) may download weights and require network/cache. Do not instantiate them in offline smoke checks unless weights are already cached.
- Training examples may need datasets, W&B/HuggingFace/S3 credentials, metric assets, and long runtime. Validate configs and data layout before launch.
- The `dream` CLI expects a combined DALLE2 checkpoint with package-expected init/model keys; trainer checkpoints are a different format.
- If `clip-anytorch` fails because `pkg_resources` is missing, install `setuptools<81` and rerun `python -m pip check`.

## Root References

- Repository snapshot and refresh baseline: [references/repo-provenance.md](references/repo-provenance.md).
- Cross-cutting install/import/backend troubleshooting: [references/troubleshooting.md](references/troubleshooting.md).
- Router metadata used by managed import: [references/repo-routing-metadata.json](references/repo-routing-metadata.json).

## Self-Containment Rule

Use the bundled references and scripts in this skill tree. Do not require future agents to open, run, or copy scripts, docs, configs, tests, or examples from an original DALLE2-pytorch checkout. When a user wants to train in a checkout intentionally, treat that checkout as their working project, not as this skill's required evidence source.

From VectorSpaceLab/AREX-Skill, MIT.

More from VectorSpaceLab/AREX-Skill