Colab notebook
Colab for this model doesnt work anymore. (Because of PyTorch version?)
AssertionError Traceback (most recent call last)
/tmp/ipython-input-773422404.py in <cell line: 0>()
----> 1 trainer_stats = trainer.train()
36 frames
/tmp/torchinductor_root/ks/cksugjoiuz2kncblvgqnbp52nhfnseufzktx6lphaxo5soadns3h.py in call(args)
121 assert_alignment(buf7, 16, 'torch.ops.bitsandbytes.int8_mixed_scaled_mm.default')
122 buf8 = buf6[1]
--> 123 assert_size_stride(buf8, (u1, ), (1, ), 'torch.ops.bitsandbytes.int8_mixed_scaled_mm.default')
124 assert_alignment(buf8, 16, 'torch.ops.bitsandbytes.int8_mixed_scaled_mm.default')
125 if not (u1 >= 0):
AssertionError: wrong number of dimensions2 for op: torch.ops.bitsandbytes.int8_mixed_scaled_mm.default