KaiyueWen commited on
Commit
e1a7fae
·
verified ·
1 Parent(s): bbba26d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -53,10 +53,13 @@ The model was trained on:
53
 
54
  ## Usage
55
 
 
 
56
  ```python
57
  import torch
58
  from transformers import AutoTokenizer
59
- from model import ScalingLawForecaster # Make sure to import the model class
 
60
 
61
  # Load model
62
  model = ScalingLawForecaster(
 
53
 
54
  ## Usage
55
 
56
+ The `ScalingLawForecaster` class can be found in the [GitHub repository](https://github.com/zhqwqwq/Configuration-to-Performance-Scaling-Law).
57
+
58
  ```python
59
  import torch
60
  from transformers import AutoTokenizer
61
+ # Get ScalingLawForecaster from: https://github.com/zhqwqwq/Configuration-to-Performance-Scaling-Law
62
+ from model import ScalingLawForecaster
63
 
64
  # Load model
65
  model = ScalingLawForecaster(