How to use ibm-research/materials.smi-ted with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="ibm-research/materials.smi-ted")
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("ibm-research/materials.smi-ted", dtype="auto")
How to use ibm-research/materials.smi-ted with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("ibm-research/materials.smi-ted", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
The community tab is the place to discuss and collaborate with the HF community!
