Add import os to vae_hf_cs.py

[Varuni Sastry]

The "import os" is missing from https://github.com/Cerebras/modelzoo/blob/main/modelzoo/vision/pytorch/dit/checkpoint_converter/vae_hf_cs.py.