Update txt2image.py (#285)

added np alias
This commit is contained in:
Rishi Narang 2024-01-11 01:31:59 +08:00 committed by GitHub
parent 2bbe9d3bd8
commit a2bc8426f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,6 +5,7 @@ import argparse
import mlx.core as mx import mlx.core as mx
from PIL import Image from PIL import Image
from tqdm import tqdm from tqdm import tqdm
import numpy as np
from stable_diffusion import StableDiffusion from stable_diffusion import StableDiffusion