mou3az commited on
Commit
5dbb0a2
·
verified ·
1 Parent(s): 050f756

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = "ITandGeneral_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')
 
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')