diff --git a/bob/bio/face/config/baseline/afffe.py b/bob/bio/face/config/baseline/afffe.py
index 840ae58e9ffbe5bd8a455a0c6c9b292474ae4968..d46eb007efbc3f88b51eaf3455bb8fd23a7cf33c 100644
--- a/bob/bio/face/config/baseline/afffe.py
+++ b/bob/bio/face/config/baseline/afffe.py
@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import afffe_baseline
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return afffe_baseline(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return afffe_baseline(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)
 
diff --git a/bob/bio/face/config/baseline/iresnet100.py b/bob/bio/face/config/baseline/iresnet100.py
index 6148d1621903ac9b0efb03608a57605859558cef..666e6f34cb94bd306abfafe8cab4c657520da052 100644
--- a/bob/bio/face/config/baseline/iresnet100.py
+++ b/bob/bio/face/config/baseline/iresnet100.py
@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet100
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return iresnet100(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return iresnet100(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)
 
diff --git a/bob/bio/face/config/baseline/iresnet34.py b/bob/bio/face/config/baseline/iresnet34.py
index f66b2b8827e0841ba9c053e62ae8d4d11aed2ab1..d8cdf0ba9f19c1186dfdd726e9db578f9586afd6 100644
--- a/bob/bio/face/config/baseline/iresnet34.py
+++ b/bob/bio/face/config/baseline/iresnet34.py
@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet34
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return iresnet34(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return iresnet34(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)
 
diff --git a/bob/bio/face/config/baseline/iresnet50.py b/bob/bio/face/config/baseline/iresnet50.py
index ae2562423db7b5db4fea5e27a8a9f69c889713a8..5a33a73410fb144098f945a02758f09de2678339 100644
--- a/bob/bio/face/config/baseline/iresnet50.py
+++ b/bob/bio/face/config/baseline/iresnet50.py
@@ -2,14 +2,14 @@ from bob.bio.face.embeddings.pytorch import iresnet50
 from bob.bio.face.utils import lookup_config_from_database
 
 
-annotation_type, fixed_positions, _ = lookup_config_from_database(
+annotation_type, fixed_positions, memory_demanding = lookup_config_from_database(
     locals().get("database")
 )
 
 
-def load(annotation_type, fixed_positions=None):
-    return iresnet50(annotation_type, fixed_positions)
+def load(annotation_type, fixed_positions=None, memory_demanding=False):
+    return iresnet50(annotation_type, fixed_positions, memory_demanding)
 
 
-pipeline = load(annotation_type, fixed_positions)
+pipeline = load(annotation_type, fixed_positions, memory_demanding)