Use blaze linux runner

oxideai · Apr 13, 2024 · 41b67d2 · 41b67d2
1 parent 6ecad2e
commit 41b67d2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/.github/workflows/validate.yml b/.github/workflows/validate.yml
@@ -25,7 +25,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        os: [blaze/macos-14, ubuntu-latest, windows-latest]
+        os: [blaze/macos-14, blaze/ubuntu-latest, windows-latest]
         rust: [stable]
         backend: ["ndarray", "wgpu", "torch"]
         include:

diff --git a/src/transformers/clip.rs b/src/transformers/clip.rs
@@ -361,12 +361,6 @@ mod tests {
 
     #[test]
     fn test_init_text_embeddings() {
-        // This test is known to fail on WGPU using mesa on a Linux Github CI.
-        #[cfg(all(feature = "std", target_os = "linux", feature = "wgpu"))]
-        if std::env::var("CI").is_ok() {
-            return;
-        }
-
         let device = Default::default();
         let clip_config = ClipConfig::v1_5();
         let text_embeddings: ClipTextEmbeddings<TestBackend> =