Instructions to use xpanceo-team/mattergen-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use xpanceo-team/mattergen-base with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("xpanceo-team/mattergen-base", 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
| ## Usage example | |
| ### Loading | |
| ```python | |
| from pymatgen.core import Structure, Lattice | |
| import torch | |
| from crystal_diffusers.pipelines.mattergen import MatterGenPipeline | |
| pipeline = MatterGenPipeline.from_pretrained("xpanceo-team/mattergen-base", use_safetensors=False) | |
| ``` | |
| ### Setting parameters | |
| ```python | |
| device = "cuda" | |
| batch_size = 2 | |
| random_seed = 42 | |
| # deterministic generation | |
| generator = torch.Generator(device=device).manual_seed(random_seed) | |
| ``` | |
| ### Generation | |
| ```python | |
| structures = pipeline( | |
| batch_size=batch_size, | |
| device=device, | |
| generator=generator | |
| ) | |
| ``` | |