<feat> using docker sdk.

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.py text eol=lf

Dockerfile

@@ -0,0 +1,17 @@
+FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn8-runtime
+
+RUN apt-get update && apt-get install -y \
+    git build-essential python3-dev ninja-build && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN pip install --upgrade pip
+
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install -r /tmp/requirements.txt
+
+RUN pip install git+https://github.com/Dao-AILab/[email protected]#subdirectory=csrc
+
+COPY . /app
+WORKDIR /app
+
+CMD ["python", "app.py"]

README.md

@@ -3,13 +3,10 @@ title: DRA-Ctrl
 emoji: 😻
 colorFrom: green
 colorTo: gray
-sdk: gradio
-sdk_version: 5.33.2
-app_file: app.py
+sdk: docker
 pinned: false
 license: apache-2.0
 short_description: huggingface space for DRA-Ctrl.
-entrypoint: app.sh
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.sh

@@ -1,4 +0,0 @@
-#!/bin/bash
-pip install torch==2.5.1
-pip install git+https://github.com/Dao-AILab/[email protected]#subdirectory=csrc
-python app.py

models/hyvideo/transformer_hunyuan_video_i2v.py

@@ -43,7 +43,7 @@ try:
     from flash_attn import flash_attn_func, flash_attn_varlen_func
     FLASH_ATTN_AVALIABLE = True
 except:
-    FLASH_ATTN_AVALIABLE = False
+    FLASH_ATTN_AVALIABLE = True
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name