Added missing sampler params (#22)

Author: spietras

kilroy_module_pytorch_py_sdk/src/kilroy_module_pytorch_py_sdk/generator/generator.py

@@ -122,6 +122,7 @@ async def _load_saved_state(self, directory: Path) -> State:
                 Sampler,
                 category=state_dict["sampler_type"],
                 default=partial(self._build_sampler, params),
+                **params.samplers_params.get(params.sampler_type, {}),
             ),
             samplers_params=state_dict["samplers_params"],
             contexts=state_dict["contexts"],