Missing import for inference
#5
by
davide221
- opened
README.md
CHANGED
@@ -47,6 +47,7 @@ import os
|
|
47 |
import torch
|
48 |
from torch import nn
|
49 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
50 |
|
51 |
## Define the reward model function class
|
52 |
|
|
|
47 |
import torch
|
48 |
from torch import nn
|
49 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
50 |
+
from huggingface_hub import snapshot_download
|
51 |
|
52 |
## Define the reward model function class
|
53 |
|