test: use CUDA if cuda feature is enabled

Updated test configurations for Translator and Generator to use CUDA device when the `cuda` feature is enabled; defaulting to CPU otherwise. This ensures that tests are run on the appropriate device based on the feature flag.

Author: jkawamoto

src/generator.rs

@@ -242,15 +242,26 @@ impl<T: Tokenizer> Debug for Generator<T> {
 #[cfg(feature = "hub")]
 mod tests {
     use super::Generator;
-    use crate::{download_model, GenerationOptions};
+    use crate::{download_model, Config, Device, GenerationOptions};
 
     const MODEL_ID: &str = "jkawamoto/gpt2-ct2";
 
     #[test]
     #[ignore]
     fn test_generate() {
         let model_path = download_model(MODEL_ID).unwrap();
-        let g = Generator::new(&model_path, &Default::default()).unwrap();
+        let g = Generator::new(
+            &model_path,
+            &Config {
+                device: if cfg!(feature = "cuda") {
+                    Device::CUDA
+                } else {
+                    Device::CPU
+                },
+                ..Default::default()
+            },
+        )
+        .unwrap();
 
         let prompt = "CTranslate2 is a library";
         let res = g

src/translator.rs

@@ -327,15 +327,27 @@ impl<T: Tokenizer> Debug for Translator<T> {
 #[cfg(test)]
 #[cfg(feature = "hub")]
 mod tests {
-    use crate::{download_model, TranslationOptions, Translator};
+    use crate::{download_model, Config, Device, TranslationOptions, Translator};
 
     const MODEL_ID: &str = "jkawamoto/fugumt-en-ja-ct2";
 
     #[test]
     #[ignore]
     fn test_translate() {
         let model_path = download_model(MODEL_ID).unwrap();
-        let t = Translator::new(&model_path, &Default::default()).unwrap();
+        let t = Translator::new(
+            &model_path,
+            &Config {
+                device: if cfg!(feature = "cuda") {
+                    Device::CUDA
+                } else {
+                    Device::CPU
+                },
+
+                ..Default::default()
+            },
+        )
+        .unwrap();
 
         let res = t
             .translate_batch(