yujiepan commited on
Commit
8e063e3
1 Parent(s): d7f9621

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +47 -0
README.md ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: text-generation
3
+ inference: true
4
+ widget:
5
+ - text: 'Hello!'
6
+ example_title: Hello world
7
+ group: Python
8
+ library_name: transformers
9
+ ---
10
+
11
+ This model is randomly initialized, using the config from [facebook/opt-30b](https://huggingface.co/facebook/opt-30b) but with smaller size.
12
+ Note the model is in float16.
13
+
14
+ Codes:
15
+ ```python
16
+ import torch
17
+ import transformers
18
+ import os
19
+ from optimum.intel.openvino import OVModelForCausalLM
20
+
21
+ save_path = '/tmp/yujiepan/opt-tiny-random'
22
+ repo_id = 'yujiepan/opt-tiny-random'
23
+
24
+ config = transformers.AutoConfig.from_pretrained('facebook/opt-30b')
25
+ config.ffn_dim = 32
26
+ config.hidden_size = 8
27
+ config.num_attention_heads = 2
28
+ config.num_hidden_layers = 2
29
+ config.word_embed_proj_dim = 8
30
+
31
+ model = transformers.AutoModelForCausalLM.from_config(config, torch_dtype=torch.float16)
32
+ model = model.half()
33
+ model.save_pretrained(save_path)
34
+
35
+ tokenizer = transformers.AutoTokenizer.from_pretrained('facebook/opt-30b')
36
+ tokenizer.save_pretrained(save_path)
37
+
38
+ ovmodel = OVModelForCausalLM.from_pretrained(save_path, export=True)
39
+ ovmodel = ovmodel.half()
40
+ ovmodel.save_pretrained(save_path)
41
+
42
+ os.system(f'ls -alh {save_path}')
43
+
44
+ from huggingface_hub import create_repo, upload_folder
45
+ create_repo(repo_id, exist_ok=True)
46
+ upload_folder(repo_id=repo_id, folder_path=save_path)
47
+ ```