push

app.py

@@ -17,7 +17,7 @@ with init_empty_weights():
   raw_model = MossForCausalLM._from_config(config, torch_dtype=torch.float)
 raw_model.tie_weights()
 model = load_checkpoint_and_dispatch(
-  raw_model, model_path, device_map="balanced_low_0", no_split_module_classes=["MossBlock"], dtype=torch.float,
+  raw_model, checkpoint=model_path, device_map="balanced_low_0", no_split_module_classes=["MossBlock"], dtype=torch.float,
   offload_folder="offload_folder"
 )