Update

[ghstack-poisoned]

Author: Vincent Moens

tensordict/base.py

@@ -11352,6 +11352,8 @@ def _validate_key(self, key: NestedKey) -> NestedKey:
 
     @property
     def _validate_value(self):
+        if is_compiling():
+            return self._validate_value_generic
         try:
             return self._validate_value_cached
         except AttributeError: