NorGLM
/

NorGPT-369M-NO-QNLI-peft

Model card Files Files and versions Community

NorGLM commited on Mar 10, 2024

Commit

f74ee9b

·

verified ·

1 Parent(s): 3ea7df0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ import torch
 torch_device = "cuda" if torch.cuda.is_available() else "cpu"
-source_model_id = "NbAiLab/nb-gpt-j-6B"
 peft_model_id = "NorGLM/NorGPT-369M-NO-QNLI-peft"
 config = PeftConfig.from_pretrained(peft_model_id)

 torch_device = "cuda" if torch.cuda.is_available() else "cpu"
+source_model_id = "NorGLM/NorGPT-369M"
 peft_model_id = "NorGLM/NorGPT-369M-NO-QNLI-peft"
 config = PeftConfig.from_pretrained(peft_model_id)