Open ZichenMiao opened 7 months ago
Hi,
Thanks for this interesting work!!
However, I got a pure noise image when running the sample code below.
Hardware:
NVIDIA A5000
code:
import os import torch import torch.nn as nn from segmoe import SegMoEPipeline pipeline = SegMoEPipeline("segmind/SegMoE-2x1-v0", device = "cuda") prompt = "cosmic canvas, orange city background, painting of a chubby cat" negative_prompt = "nsfw, bad quality, worse quality" img = pipeline( prompt=prompt, negative_prompt=negative_prompt, height=1024, width=1024, num_inference_steps=25, guidance_scale=7.5, ).images[0] img.save("image.png")
environment:
# Name Version Build Channel _libgcc_mutex 0.1 main _openmp_mutex 5.1 1_gnu accelerate 0.29.3 pypi_0 pypi bzip2 1.0.8 h5eee18b_5 ca-certificates 2024.3.11 h06a4308_0 certifi 2024.2.2 pypi_0 pypi charset-normalizer 3.3.2 pypi_0 pypi diffusers 0.27.2 pypi_0 pypi filelock 3.13.4 pypi_0 pypi fsspec 2024.3.1 pypi_0 pypi huggingface-hub 0.22.2 pypi_0 pypi idna 3.7 pypi_0 pypi importlib-metadata 7.1.0 pypi_0 pypi jinja2 3.1.3 pypi_0 pypi ld_impl_linux-64 2.38 h1181459_1 libffi 3.3 he6710b0_2 libgcc-ng 11.2.0 h1234567_1 libgomp 11.2.0 h1234567_1 libstdcxx-ng 11.2.0 h1234567_1 libuuid 1.41.5 h5eee18b_0 markupsafe 2.1.5 pypi_0 pypi mpmath 1.3.0 pypi_0 pypi ncurses 6.4 h6a678d5_0 networkx 3.3 pypi_0 pypi numpy 1.26.4 pypi_0 pypi nvidia-cublas-cu12 12.1.3.1 pypi_0 pypi nvidia-cuda-cupti-cu12 12.1.105 pypi_0 pypi nvidia-cuda-nvrtc-cu12 12.1.105 pypi_0 pypi nvidia-cuda-runtime-cu12 12.1.105 pypi_0 pypi nvidia-cudnn-cu12 8.9.2.26 pypi_0 pypi nvidia-cufft-cu12 11.0.2.54 pypi_0 pypi nvidia-curand-cu12 10.3.2.106 pypi_0 pypi nvidia-cusolver-cu12 11.4.5.107 pypi_0 pypi nvidia-cusparse-cu12 12.1.0.106 pypi_0 pypi nvidia-nccl-cu12 2.19.3 pypi_0 pypi nvidia-nvjitlink-cu12 12.4.127 pypi_0 pypi nvidia-nvtx-cu12 12.1.105 pypi_0 pypi openssl 1.1.1w h7f8727e_0 packaging 24.0 pypi_0 pypi pillow 10.3.0 pypi_0 pypi pip 23.3.1 py310h06a4308_0 psutil 5.9.8 pypi_0 pypi python 3.10.6 haa1d7c7_1 pyyaml 6.0.1 pypi_0 pypi readline 8.2 h5eee18b_0 regex 2024.4.16 pypi_0 pypi requests 2.31.0 pypi_0 pypi safetensors 0.4.3 pypi_0 pypi segmoe 0.0.4 pypi_0 pypi setuptools 68.2.2 py310h06a4308_0 sqlite 3.41.2 h5eee18b_0 sympy 1.12 pypi_0 pypi tk 8.6.12 h1ccaba5_0 tokenizers 0.19.1 pypi_0 pypi torch 2.2.2 pypi_0 pypi tqdm 4.66.2 pypi_0 pypi transformers 4.40.0 pypi_0 pypi triton 2.2.0 pypi_0 pypi typing-extensions 4.11.0 pypi_0 pypi tzdata 2024a h04d1e81_0 urllib3 2.2.1 pypi_0 pypi wheel 0.41.2 py310h06a4308_0 xz 5.4.6 h5eee18b_0 zipp 3.18.1 pypi_0 pypi zlib 1.2.13 h5eee18b_0
Image I got:
Hi,
Thanks for this interesting work!!
However, I got a pure noise image when running the sample code below.
Hardware:
code:
environment:
Image I got: