not-lain commited on
Commit
5ec5990
1 Parent(s): 4a02b16

update readme and requirements

Browse files
Files changed (2) hide show
  1. README.md +10 -0
  2. requirements.txt +2 -0
README.md CHANGED
@@ -39,6 +39,16 @@ The training setup was `4xA100's 80GB` and took ~6 hours to pretrain and ~13 hou
39
  | ![extreme_ironing](examples/extreme_ironing.jpg) | **What's funny about this image?**<br>The image is quite humorous as it depicts a man ironing clothes on the back of a yellow taxi cab. This is not a typical sight you'd expect to see in everyday life. |
40
  ---
41
 
 
 
 
 
 
 
 
 
 
 
42
 
43
  ## Training:
44
  We will release the training code in some time.
 
39
  | ![extreme_ironing](examples/extreme_ironing.jpg) | **What's funny about this image?**<br>The image is quite humorous as it depicts a man ironing clothes on the back of a yellow taxi cab. This is not a typical sight you'd expect to see in everyday life. |
40
  ---
41
 
42
+ ## Loading the model
43
+
44
+ ```
45
+ pip install -qr https://huggingface.co/Tensoic/Cerule-v0.1/resolve/main/requirements.txt
46
+ ```
47
+
48
+ ```python
49
+ from transformers import AutoModelForCausalLM
50
+ model = AutoModelForCausalLM.from_pretrained("Tensoic/Cerule-v0.1", trust_remote_code=True)
51
+ ```
52
 
53
  ## Training:
54
  We will release the training code in some time.
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ flash_attn
2
+ transformers>=4.39.1