pip install transformers | |
import torch | |
from transformers import TemporalNetStableDiffusionModel, TemporalNetStableDiffusionConfig, AutoTokenizer | |
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/temporalnet_stablediffusion_enwik8") | |
config = TemporalNetStableDiffusionConfig.from_pretrained("EleutherAI/temporalnet_stablediffusion_enwik8") | |
model = TemporalNetStableDiffusionModel.from_pretrained("EleutherAI/temporalnet_stablediffusion_enwik8", config=config) | |