LeoBorai commited on
Commit
3fdf998
·
1 Parent(s): 470fea7

feat: train larger model

Browse files
Files changed (2) hide show
  1. assets/model.json +0 -0
  2. model/src/model/gpu.rs +1 -2
assets/model.json CHANGED
The diff for this file is too large to render. See raw diff
 
model/src/model/gpu.rs CHANGED
@@ -1,7 +1,6 @@
1
  //! Metal-accelerated Batch Gradient Descent, using `candle` to run the
2
  //! matmul-heavy training loop on the local machine's GPU (e.g. Apple
3
- //! Silicon). Only built when the `training` feature is enabled - the
4
- //! deployed server never links `candle`.
5
 
6
  use candle_core::{DType, Device, Tensor};
7
  use ndarray::Array2;
 
1
  //! Metal-accelerated Batch Gradient Descent, using `candle` to run the
2
  //! matmul-heavy training loop on the local machine's GPU (e.g. Apple
3
+ //! Silicon).
 
4
 
5
  use candle_core::{DType, Device, Tensor};
6
  use ndarray::Array2;