diff --git a/setup.py b/setup.py
index 298c033..bb8fbbe 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='ttools',
-    version='0.7.93',
+    version='0.7.94',
     packages=find_packages(),
     install_requires=[
         # list your dependencies here
diff --git a/ttools/models.py b/ttools/models.py
index 2e0c28e..2aab27a 100644
--- a/ttools/models.py
+++ b/ttools/models.py
@@ -902,7 +902,9 @@ class LibraryTradingModel:
                         model.set_params(num_class=self.config.n_classes)
                 else:
                     model = XGBRegressor(**params)
-                
+
+                model = set_gpu_params(model)
+
                 # Handle class imbalance for binary classification
                 if self.config.n_classes == 2:
                     n_0 = sum(y_fold_train == 0)