Spaces:

EvanTHU
/

MotionLLM

Running on Zero

EvanTHU commited on Jun 6, 2024

Commit

e53f6f9

verified ·

1 Parent(s): edcbefc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -508,7 +508,7 @@ quantize = None
 t0 = time.time()
 mlp_path = args.mlp_path
-pretrained_checkpoint_mlp = torch.load(mlp_path)
 with EmptyInitOnDevice(
     device=fabric.device, dtype=dtype, quantization_mode=quantize

 t0 = time.time()
 mlp_path = args.mlp_path
+pretrained_checkpoint_mlp = torch.load(mlp_path, map_location='cpu')
 with EmptyInitOnDevice(
     device=fabric.device, dtype=dtype, quantization_mode=quantize