Update README.md
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ tags:
|
|
| 40 |
from peft import PeftModel, PeftConfig
|
| 41 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 42 |
HUGGING_FACE_USER_NAME = "mou3az"
|
| 43 |
-
model_name = "
|
| 44 |
peft_model_id = f"{HUGGING_FACE_USER_NAME}/{model_name}"
|
| 45 |
config = PeftConfig.from_pretrained(peft_model_id)
|
| 46 |
model = AutoModelForSeq2SeqLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=False, device_map='auto')
|
|
|
|
| 40 |
from peft import PeftModel, PeftConfig
|
| 41 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 42 |
HUGGING_FACE_USER_NAME = "mou3az"
|
| 43 |
+
model_name = "IT-General_Question-Generation "
|
| 44 |
peft_model_id = f"{HUGGING_FACE_USER_NAME}/{model_name}"
|
| 45 |
config = PeftConfig.from_pretrained(peft_model_id)
|
| 46 |
model = AutoModelForSeq2SeqLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=False, device_map='auto')
|