Spaces:
Sleeping
Sleeping
feat: train larger model
Browse files- assets/model.json +0 -0
- 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).
|
| 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;
|