directly set device index the first time to avoid initializing unused…

… gpu (OpenNMT#1633)
Dagamies · Mar 1, 2024 · 1427722 · 1427722
1 parent 63c1ae5
commit 1427722
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models/model.cc b/src/models/model.cc
@@ -423,7 +423,7 @@ namespace ctranslate2 {
 
       {
         // Check that the device and device index are valid.
-        ScopedDeviceSetter(device, device_index);
+        set_device_index(device, device_index);
       }
 
       std::unique_ptr<std::istream> model_file_ptr = model_reader.get_required_file(binary_file,