KaiyueWen commited on
Commit
58bac15
·
verified ·
1 Parent(s): 867babb

Upload README.md with huggingface_hub

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