For me, above code is working fine. This is output which I got on running above mentioned code. Code Output
VM resources are: 14GB RAM, 28GB Disk Falcon 7B Falcon requires minimum 14 GB RAM. For reference: https://huggingface.co/tiiuae/falcon-7b/discussions/2