OptimizerStudy
/

NCPL-intermediate

Feature Extraction

scaling_law_forecaster

performance-prediction

configuration-to-performance

Model card Files Files and versions

KaiyueWen commited on Feb 17

Commit

e1a7fae

·

verified ·

1 Parent(s): bbba26d

Upload README.md with huggingface_hub

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -53,10 +53,13 @@ The model was trained on:
 ## Usage
 ```python
 import torch
 from transformers import AutoTokenizer
-from model import ScalingLawForecaster  # Make sure to import the model class
 # Load model
 model = ScalingLawForecaster(

 ## Usage
+The `ScalingLawForecaster` class can be found in the [GitHub repository](https://github.com/zhqwqwq/Configuration-to-Performance-Scaling-Law).
 ```python
 import torch
 from transformers import AutoTokenizer
+# Get ScalingLawForecaster from: https://github.com/zhqwqwq/Configuration-to-Performance-Scaling-Law
+from model import ScalingLawForecaster
 # Load model
 model = ScalingLawForecaster(