Instructions to use Vivek/gptneo_hellaswag with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Vivek/gptneo_hellaswag with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForMultipleChoice tokenizer = AutoTokenizer.from_pretrained("Vivek/gptneo_hellaswag") model = AutoModelForMultipleChoice.from_pretrained("Vivek/gptneo_hellaswag", device_map="auto") - Notebooks
- Google Colab
- Kaggle
changees
Browse files- src/test_hellaswag.py +1 -1
src/test_hellaswag.py
CHANGED
|
@@ -75,7 +75,7 @@ for idx,batch in enumerate(glue_test_data_loader(input_rng, test_dataset, total_
|
|
| 75 |
restored_output.append(final_output)
|
| 76 |
|
| 77 |
finall=pd.DataFrame({'predictions':restored_output,'permutation':list1})
|
| 78 |
-
finall.to_csv('.predictions.csv')
|
| 79 |
|
| 80 |
|
| 81 |
|
|
|
|
| 75 |
restored_output.append(final_output)
|
| 76 |
|
| 77 |
finall=pd.DataFrame({'predictions':restored_output,'permutation':list1})
|
| 78 |
+
finall.to_csv('./predictions.csv')
|
| 79 |
|
| 80 |
|
| 81 |
|