ci: add text-to-image example

huggingface · Jul 23, 2024 · c985f22 · c985f22
1 parent 9594236
commit c985f22
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/.github/workflows/linux-examples.yml b/.github/workflows/linux-examples.yml
@@ -51,7 +51,7 @@ jobs:
                 --weights $w --activations $a --device ${{ matrix.device }}; \
             done; \
           done
-      - name: Run OWL detection examples
+      - name: Run OWL detection example
         run: |
           for w in int4 int8 float8; do \
             python examples/vision/object-detection/quantize_owl_model.py \
@@ -67,4 +67,10 @@ jobs:
                 --weights $w --activations $a --device ${{ matrix.device }}; \
             done; \
           done
-
+      - name: Run text-to-image example
+        if: ${{ matrix.device == 'cuda'}}
+        run: |
+          for w in int4 int8 fp8; do \
+            python examples/vision/text-to-image/quantize_pixart_sigma.py \
+              --qtype $w --device ${{ matrix.device }}; \
+          done