jeanflop commited on
Commit
dcf0792
1 Parent(s): 1a4050f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -19,7 +19,7 @@ This model lora weight has been finetune on french OCR dataset. The architecture
19
  ```
20
  * **Load and merge adaptaters in 8Bit** (recommanded)
21
 
22
- ```
23
  import torch
24
  from peft import PeftModel, PeftConfig
25
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer,BitsAndBytesConfig
@@ -42,12 +42,12 @@ print("Peft model loaded")
42
 
43
  Add your text
44
 
45
- ```
46
  inputs=f"""
47
  Fix text : {text}"""
48
  ```
49
  Run
50
- ```
51
  peft_model.config.max_length=512
52
  peft_tokenizer.model_max_length=512
53
  inputs = peft_tokenizer(inputs, return_tensors="pt")
 
19
  ```
20
  * **Load and merge adaptaters in 8Bit** (recommanded)
21
 
22
+ ```python
23
  import torch
24
  from peft import PeftModel, PeftConfig
25
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer,BitsAndBytesConfig
 
42
 
43
  Add your text
44
 
45
+ ```python
46
  inputs=f"""
47
  Fix text : {text}"""
48
  ```
49
  Run
50
+ ```python
51
  peft_model.config.max_length=512
52
  peft_tokenizer.model_max_length=512
53
  inputs = peft_tokenizer(inputs, return_tensors="pt")