Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c83683fdd1 | |||
| 6d6d19d7fc |
@@ -25,17 +25,17 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Set up Docker Buildx
|
- name: Set up Docker Buildx
|
||||||
uses: docker/setup-buildx-action@v1
|
uses: docker/setup-buildx-action@v1
|
||||||
|
|
||||||
- name: Check out code
|
- name: Check out code
|
||||||
uses: actions/checkout@v3
|
uses: actions/checkout@v3
|
||||||
|
|
||||||
- name: Find Changed Dockerfiles
|
- name: Find Changed Dockerfiles
|
||||||
id: file_changes
|
id: file_changes
|
||||||
uses: jitterbit/get-changed-files@v1
|
uses: jitterbit/get-changed-files@v1
|
||||||
with:
|
with:
|
||||||
format: 'space-delimited'
|
format: 'space-delimited'
|
||||||
token: ${{ secrets.GITHUB_TOKEN }}
|
token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|
||||||
- name: Build Changed Docker Images
|
- name: Build Changed Docker Images
|
||||||
run: |
|
run: |
|
||||||
CHANGED_FILES="${{ steps.file_changes.outputs.all }}"
|
CHANGED_FILES="${{ steps.file_changes.outputs.all }}"
|
||||||
@@ -52,7 +52,7 @@ jobs:
|
|||||||
build-and-push-docker-images:
|
build-and-push-docker-images:
|
||||||
runs-on: [ self-hosted, intel-cpu, 8-cpu, ci ]
|
runs-on: [ self-hosted, intel-cpu, 8-cpu, ci ]
|
||||||
if: github.event_name != 'pull_request'
|
if: github.event_name != 'pull_request'
|
||||||
|
|
||||||
permissions:
|
permissions:
|
||||||
contents: read
|
contents: read
|
||||||
packages: write
|
packages: write
|
||||||
@@ -69,7 +69,6 @@ jobs:
|
|||||||
- diffusers-flax-tpu
|
- diffusers-flax-tpu
|
||||||
- diffusers-onnxruntime-cpu
|
- diffusers-onnxruntime-cpu
|
||||||
- diffusers-onnxruntime-cuda
|
- diffusers-onnxruntime-cuda
|
||||||
- diffusers-doc-builder
|
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout repository
|
- name: Checkout repository
|
||||||
|
|||||||
@@ -1,50 +0,0 @@
|
|||||||
FROM ubuntu:20.04
|
|
||||||
LABEL maintainer="Hugging Face"
|
|
||||||
LABEL repository="diffusers"
|
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive
|
|
||||||
|
|
||||||
RUN apt-get -y update \
|
|
||||||
&& apt-get install -y software-properties-common \
|
|
||||||
&& add-apt-repository ppa:deadsnakes/ppa
|
|
||||||
|
|
||||||
RUN apt install -y bash \
|
|
||||||
build-essential \
|
|
||||||
git \
|
|
||||||
git-lfs \
|
|
||||||
curl \
|
|
||||||
ca-certificates \
|
|
||||||
libsndfile1-dev \
|
|
||||||
python3.10 \
|
|
||||||
python3-pip \
|
|
||||||
libgl1 \
|
|
||||||
python3.10-venv && \
|
|
||||||
rm -rf /var/lib/apt/lists
|
|
||||||
|
|
||||||
# make sure to use venv
|
|
||||||
RUN python3.10 -m venv /opt/venv
|
|
||||||
ENV PATH="/opt/venv/bin:$PATH"
|
|
||||||
|
|
||||||
# pre-install the heavy dependencies (these can later be overridden by the deps from setup.py)
|
|
||||||
RUN python3.10 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
|
|
||||||
python3.10 -m uv pip install --no-cache-dir \
|
|
||||||
torch \
|
|
||||||
torchvision \
|
|
||||||
torchaudio \
|
|
||||||
invisible_watermark \
|
|
||||||
--extra-index-url https://download.pytorch.org/whl/cpu && \
|
|
||||||
python3.10 -m uv pip install --no-cache-dir \
|
|
||||||
accelerate \
|
|
||||||
datasets \
|
|
||||||
hf-doc-builder \
|
|
||||||
huggingface-hub \
|
|
||||||
Jinja2 \
|
|
||||||
librosa \
|
|
||||||
numpy \
|
|
||||||
scipy \
|
|
||||||
tensorboard \
|
|
||||||
transformers \
|
|
||||||
matplotlib \
|
|
||||||
setuptools==69.5.1
|
|
||||||
|
|
||||||
CMD ["/bin/bash"]
|
|
||||||
@@ -31,6 +31,7 @@ from ..utils import (
|
|||||||
is_transformers_available,
|
is_transformers_available,
|
||||||
is_xformers_available,
|
is_xformers_available,
|
||||||
)
|
)
|
||||||
|
from ..utils.testing_utils import get_python_version
|
||||||
from . import BaseDiffusersCLICommand
|
from . import BaseDiffusersCLICommand
|
||||||
|
|
||||||
|
|
||||||
@@ -105,6 +106,11 @@ class EnvironmentCommand(BaseDiffusersCLICommand):
|
|||||||
|
|
||||||
xformers_version = xformers.__version__
|
xformers_version = xformers.__version__
|
||||||
|
|
||||||
|
if get_python_version() >= (3, 10):
|
||||||
|
platform_info = f"{platform.freedesktop_os_release().get('PRETTY_NAME', None)} - {platform.platform()}"
|
||||||
|
else:
|
||||||
|
platform_info = platform.platform()
|
||||||
|
|
||||||
is_notebook_str = "Yes" if is_notebook() else "No"
|
is_notebook_str = "Yes" if is_notebook() else "No"
|
||||||
|
|
||||||
is_google_colab_str = "Yes" if is_google_colab() else "No"
|
is_google_colab_str = "Yes" if is_google_colab() else "No"
|
||||||
@@ -152,7 +158,7 @@ class EnvironmentCommand(BaseDiffusersCLICommand):
|
|||||||
|
|
||||||
info = {
|
info = {
|
||||||
"🤗 Diffusers version": version,
|
"🤗 Diffusers version": version,
|
||||||
"Platform": f"{platform.freedesktop_os_release().get('PRETTY_NAME', None)} - {platform.platform()}",
|
"Platform": platform_info,
|
||||||
"Running on a notebook?": is_notebook_str,
|
"Running on a notebook?": is_notebook_str,
|
||||||
"Running on Google Colab?": is_google_colab_str,
|
"Running on Google Colab?": is_google_colab_str,
|
||||||
"Python version": platform.python_version(),
|
"Python version": platform.python_version(),
|
||||||
|
|||||||
@@ -178,7 +178,7 @@ class StableVideoDiffusionPipeline(DiffusionPipeline):
|
|||||||
feature_extractor=feature_extractor,
|
feature_extractor=feature_extractor,
|
||||||
)
|
)
|
||||||
self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
|
self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
|
||||||
self.video_processor = VideoProcessor(do_resize=True, vae_scale_factor=self.vae_scale_factor)
|
self.video_processor = VideoProcessor(do_resize=False, vae_scale_factor=self.vae_scale_factor)
|
||||||
|
|
||||||
def _encode_image(
|
def _encode_image(
|
||||||
self,
|
self,
|
||||||
|
|||||||
Reference in New Issue
Block a user