[AudioLM] Graph break: 'inline in skipfiles: Random.randrange'

[ezyang]

🐛 Describe the bug

Discovered while compiling lucidrains/audiolm-pytorch

[2024-03-06 11:44:11,192] [25/0] torch._dynamo.symbolic_convert.__graph_breaks: [DEBUG] Graph break: 'inline in skipfiles: Random.randrange | randrange /home/ezyang/local/miniconda3-test/envs/audiolm/lib/python3.10/random.py, skipped according skipfiles.SKIP_DIRS' from user code at:
[2024-03-06 11:44:11,192] [25/0] torch._dynamo.symbolic_convert.__graph_breaks: [DEBUG]   File "/home/ezyang/local/miniconda3-test/envs/audiolm/lib/python3.10/site-packages/vector_quantize_pytorch-1.14.1-py3.10.egg/vector_quantize_pytorch/residual_vq.py", line 155, in resume_in_forward
[2024-03-06 11:44:11,192] [25/0] torch._dynamo.symbolic_convert.__graph_breaks: [DEBUG]     rand_quantize_dropout_index = rand.randrange(self.quantize_dropout_cutoff_index, num_quant)
[2024-03-06 11:44:11,192] [25/0] torch._dynamo.symbolic_convert.__graph_breaks: [DEBUG] 

Full repro code: gist.github.com/ezyang/64c24c9fc5529f3afed4ee4266f6adc5

Versions

main

cc @chauhang @penguinwu @voznesenskym @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @amjames @msaroufim @bdhirsh @anijain2305 @zou3519

[williamwen42]

This specific graph break is fixed, but there is a different graph break encountered (see #133725)