abumafrim commited on
Commit
858e6fc
·
verified ·
1 Parent(s): cc6cbcd

Update example model/adapter paths to dsfsi namespace

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -18,7 +18,7 @@ pipeline_tag: translation
18
 
19
  # gemma_2_9b_it-lora-r4-amh-eng
20
 
21
- [![Model on HF](https://huggingface.co/datasets/huggingface/badges/raw/main/model-on-hf-sm.svg)](https://huggingface.co/AfriScience-MT/gemma_2_9b_it-lora-r4-amh-eng)
22
 
23
  This is a **LoRA adapter** for the AfriScience-MT project, enabling efficient scientific machine translation for African languages.
24
 
@@ -77,7 +77,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
77
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
78
 
79
  # Load LoRA adapter
80
- adapter_name = "AfriScience-MT/gemma_2_9b_it-lora-r4-amh-eng"
81
  model = PeftModel.from_pretrained(base_model, adapter_name)
82
  model.eval()
83
 
@@ -115,7 +115,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
115
  device_map="auto",
116
  torch_dtype=torch.bfloat16,
117
  )
118
- model = PeftModel.from_pretrained(base_model, "AfriScience-MT/gemma_2_9b_it-lora-r4-amh-eng")
119
  ```
120
 
121
  ## Training Details
 
18
 
19
  # gemma_2_9b_it-lora-r4-amh-eng
20
 
21
+ [![Model on HF](https://huggingface.co/datasets/huggingface/badges/raw/main/model-on-hf-sm.svg)](https://huggingface.co/dsfsi/gemma_2_9b_it-lora-r4-amh-eng)
22
 
23
  This is a **LoRA adapter** for the AfriScience-MT project, enabling efficient scientific machine translation for African languages.
24
 
 
77
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
78
 
79
  # Load LoRA adapter
80
+ adapter_name = "dsfsi/gemma_2_9b_it-lora-r4-amh-eng"
81
  model = PeftModel.from_pretrained(base_model, adapter_name)
82
  model.eval()
83
 
 
115
  device_map="auto",
116
  torch_dtype=torch.bfloat16,
117
  )
118
+ model = PeftModel.from_pretrained(base_model, "dsfsi/gemma_2_9b_it-lora-r4-amh-eng")
119
  ```
120
 
121
  ## Training Details