cell 1 code: -------------------------------------------- #@title # [minDALL-E](https://github.com/kakaobrain/minDALL-E) Inferencing #modified version of mega b notebook https://colab.research.google.com/drive/1Gg7-c7LrUTNfQ-Fk-BVNCe9kvedZZsAh?usp=sharing
import os from IPython.display import clear_output !git clone -q https://github.com/kakaobrain/minDALL-E.git %cd minDALL-E/ !pip install -q tokenizers>=0.10.2 !pip install -q pyflakes>=2.2.0 !pip install -q tqdm>=4.46.0 !pip install -q pytorch-lightning>=1.5 !pip install -q einops !pip install -q omegaconf !pip install -q git+https://github.com/openai/CLIP.git clear_output() print("Downloading model...") device = 'cuda:0' from dalle.utils.utils import set_seed from dalle.models import Dalle import clip import math %matplotlib inline model = Dalle.from_pretrained('minDALL-E/1.3B') # This will automatically download the pretrained model. model.to(device=device) model_clip, preprocess_clip = clip.load("ViT-B/32", device=device) model_clip.to(device=device) clear_output() ------------------------------------------ Artificial General Intelligence List: AGI Permalink: https://agi.topicbox.com/groups/agi/T4ae89526c13077bd-M22c34bb8a20bebbad6b2f39e Delivery options: https://agi.topicbox.com/groups/agi/subscription
