up

README.md

@@ -14,12 +14,12 @@ inference: false
 ---
 
 <p align="center" width="100%">
-<img src="https://huggingface.co/bofenghuang/vigogne-lora-bloom-7b1/resolve/main/vigogne_logo.png" alt="Vigogne" style="width: 40%; min-width: 300px; display: block; margin: auto;">
+<img src="https://huggingface.co/bofenghuang/vigogne-instruct-bloom-7b1/resolve/main/vigogne_logo.png" alt="Vigogne" style="width: 40%; min-width: 300px; display: block; margin: auto;">
 </p>
 
-# Vigogne-LoRA-BLOOM-7b1: A French Instruct BLOOM Model
+# Vigogne-instruct-bloom-7b1: A French Instruction-following BLOOM Model
 
-Vigogne-LoRA-BLOOM-7b1 is a [bigscience/bloom-7b1](https://huggingface.co/bigscience/bloom-7b1) model fine-tuned on the translated [Stanford Alpaca](https://github.com/tatsu-lab/stanford_alpaca) dataset to follow the 🇫🇷 French instructions.
+Vigogne-instruct-bloom-7b1 is a [bigscience/bloom-7b1](https://huggingface.co/bigscience/bloom-7b1) model fine-tuned to follow the 🇫🇷 French instructions.
 
 For more information, please visit the Github repo: https://github.com/bofenghuang/vigogne
 
@@ -33,23 +33,23 @@ This repo only contains the low-rank adapter. In order to access the complete mo
 from peft import PeftModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained("bigscience/bloom-7b1"")
+base_model_name_or_path = "bigscience/bloom-7b1"
+lora_model_name_or_path = "bofenghuang/vigogne-instruct-bloom-7b1"
+
+tokenizer = AutoTokenizer.from_pretrained(base_model_name_or_path, padding_side="right", use_fast=False))
 model = AutoModelForCausalLM.from_pretrained(
-    "bigscience/bloom-7b1"",
+    base_model_name_or_path,
     load_in_8bit=True,
+    torch_dtype=torch.float16,
     device_map="auto",
 )
-model = PeftModel.from_pretrained(model, "bofenghuang/vigogne-lora-bloom-7b1")
+model = PeftModel.from_pretrained(model, lora_model_name_or_path)
 ```
 
 You can infer this model by using the following Google Colab Notebook.
 
-<a href="https://colab.research.google.com/github/bofenghuang/vigogne/blob/main/infer.ipynb" target="_blank"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
+<a href="https://colab.research.google.com/github/bofenghuang/vigogne/blob/main/notebooks/infer_chat.ipynb" target="_blank"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
 
 ## Limitations
 
 Vigogne is still under development, and there are many limitations that have to be addressed. Please note that it is possible that the model generates harmful or biased content, incorrect information or generally unhelpful answers.
-
-## Next Steps
-
-- Add output examples

runs/Mar26_00-12-34_koios.zaion.ai/1679785954.4135473/events.out.tfevents.1679785954.koios.zaion.ai

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:06cabb35dd7abcaf75a82b19399ea7d6c5bbddaa10a0d1bbdd7352edc6af9ccc
-size 5569

runs/Mar26_00-12-34_koios.zaion.ai/events.out.tfevents.1679785954.koios.zaion.ai

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d662bcd78d8c45c8c269764975563ddd39bd0980484741a4529d1f1af79d520d
-size 13022