View file src/colab/huggingface_opendalle.py - Download

# -*- coding: utf-8 -*-
"""huggingface-opendalle.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1a_0Gz1y9e2Bh5TTxQcD4jTdKWOnxl9tN

https://huggingface.co/dataautogpt3/OpenDalleV1.1
"""

!pip install diffusers
!pip install accelerate

from diffusers import AutoPipelineForText2Image
import torch

from IPython.display import display

pipeline = AutoPipelineForText2Image.from_pretrained('dataautogpt3/OpenDalleV1.1', torch_dtype=torch.float16).to('cuda')

image = pipeline('black fluffy gorgeous dangerous cat animal creature, large orange eyes, big fluffy ears, piercing gaze, full moon, dark ambiance, best quality, extremely detailed').images[0]
display(image)