[BugFix] Fix ModelBasedEnvBase for torch.compile compatibility#3299

vmoens merged 2 commits intogh/vmoens/180/basepytorch/rl:gh/vmoens/180/basefrom

gh/vmoens/180/headpytorch/rl:gh/vmoens/180/headCopy head branch name to clipboard