huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Trying to fix flakyness by making hub_2 and hub_3 serial tests (potential issue on mingw with mmap).

Browse Source
This commit is contained in:
Nicolas Patry
2023-12-29 11:13:33 +01:00
parent a6bd0b47a5
commit 5b12fbb143
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/rust-ci.yml vendored
View File

@ -40,8 +40,6 @@ jobs:
profile: minimal profile: minimal
toolchain: ${{ matrix.rust }} toolchain: ${{ matrix.rust }}
override: true override: true
- name: Setup tmate session
uses: mxschmitt/action-tmate@v3
- uses: actions-rs/cargo@v1 - uses: actions-rs/cargo@v1
with: with:
command: test command: test

8
candle-book/src/lib.rs
View File

@ -43,11 +43,11 @@ let mmap = unsafe { Mmap::map(&file).unwrap() };
let weights = candle::safetensors::load_buffer(&mmap[..], &Device::Cpu).unwrap(); let weights = candle::safetensors::load_buffer(&mmap[..], &Device::Cpu).unwrap();
// ANCHOR_END: book_hub_2 // ANCHOR_END: book_hub_2
assert_eq!(weights.len(), 206); assert_eq!(weights.len(), 206);
} // }
#[rustfmt::skip] // #[rustfmt::skip]
#[test] // #[test]
fn book_hub_3() { // fn book_hub_3() {
// ANCHOR: book_hub_3 // ANCHOR: book_hub_3
use candle::{DType, Device, Tensor}; use candle::{DType, Device, Tensor};
use hf_hub::api::sync::Api; use hf_hub::api::sync::Api;