berkeley-nest
/

Starling-RM-7B-alpha

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

davide221 commited on Jan 13

Commit

4835c87

•

1 Parent(s): 12875de

Missing import for inference

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -47,6 +47,7 @@ import os
 import torch
 from torch import nn
 from transformers import AutoModelForCausalLM, AutoTokenizer
 ## Define the reward model function class

 import torch
 from torch import nn
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import snapshot_download
 ## Define the reward model function class