Fix FSDP NB

[kshitij12345]

Currently the NB fails with

[rank0]:   File "/opt/pytorch/lightning-thunder/notebooks/dev_tutorials/thunder_fsdp_simple_example.py", line 49, in <module>
[rank0]:     cmodel = thunder.jit(thunder.distributed.fsdp(model))
[rank0]:                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]:   File "/opt/pytorch/lightning-thunder/thunder/distributed/__init__.py", line 435, in fsdp
[rank0]:     raise TypeError(
[rank0]: TypeError: Expected `model` to be of type `thunder.core.module.ThunderModule`, but got Sequential

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[t-vi]

This does the trick for fixing notebook ci.