frankwong2001 commited on
Commit
bca02de
·
verified ·
1 Parent(s): 3b391bc

Update ML Intern artifact metadata

Browse files
Files changed (1) hide show
  1. README.md +21 -0
README.md CHANGED
@@ -4,6 +4,7 @@ license: apache-2.0
4
  base_model: distilbert/distilbert-base-uncased
5
  tags:
6
  - generated_from_trainer
 
7
  model-index:
8
  - name: distilbert-imdb-tiny-cpu
9
  results: []
@@ -56,3 +57,23 @@ The following hyperparameters were used during training:
56
  - Pytorch 2.11.0+cu130
57
  - Datasets 4.8.5
58
  - Tokenizers 0.22.2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  base_model: distilbert/distilbert-base-uncased
5
  tags:
6
  - generated_from_trainer
7
+ - ml-intern
8
  model-index:
9
  - name: distilbert-imdb-tiny-cpu
10
  results: []
 
57
  - Pytorch 2.11.0+cu130
58
  - Datasets 4.8.5
59
  - Tokenizers 0.22.2
60
+
61
+ <!-- ml-intern-provenance -->
62
+ ## Generated by ML Intern
63
+
64
+ This model repository was generated by [ML Intern](https://github.com/huggingface/ml-intern), an agent for machine learning research and development on the Hugging Face Hub.
65
+
66
+ - Try ML Intern: https://smolagents-ml-intern.hf.space
67
+ - Source code: https://github.com/huggingface/ml-intern
68
+
69
+ ## Usage
70
+
71
+ ```python
72
+ from transformers import AutoModelForCausalLM, AutoTokenizer
73
+
74
+ model_id = 'frankwong2001/distilbert-imdb-tiny-cpu'
75
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
76
+ model = AutoModelForCausalLM.from_pretrained(model_id)
77
+ ```
78
+
79
+ For non-causal architectures, replace `AutoModelForCausalLM` with the appropriate `AutoModel` class.