From 261684f92eed434480281e0584e5c74c9d25aef6 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 21 Dec 2022 16:49:30 -0300
Subject: [PATCH 1/3] Update README.md

---
 README.md | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index ac775212..f8029960 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,8 @@
 # text-generation-webui
-A gradio webui for running large language models locally. Supports gpt-j-6B, gpt-neox-20b, opt, galactica, and many others.
+
+A gradio webui for running large language models locally. Supports gpt-j-6B, gpt-neox-20b, opt, galactica, and many others. 
+
+Its goal is to become the [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) of text generation.
 
 ![webui screenshot](https://github.com/oobabooga/text-generation-webui/raw/main/webui.png)
 
@@ -7,6 +10,8 @@ A gradio webui for running large language models locally. Supports gpt-j-6B, gpt
 
     conda env create -f environment.yml
 
+This installs the CUDA version of pytorch, which assumes that you have a NVIDIA GPU. If you want to run this on an AMD GPU, you should install the ROCM version of pytorch instead.
+
 ## Downloading models
 
 Models should be placed under `models/model-name`.
@@ -24,6 +29,18 @@ The files that you need to download and put under `models/gpt-j-6B` are the json
 * Torrent: [16-bit](https://archive.org/details/gpt4chan_model_float16) / [32-bit](https://archive.org/details/gpt4chan_model)
 * Direct download: [16-bit](https://theswissbay.ch/pdf/_notpdf_/gpt4chan_model_float16/) / [32-bit](https://theswissbay.ch/pdf/_notpdf_/gpt4chan_model/)
 
+## Converting to pytorch
+
+This webui allows you to switch between different models on the fly, so it must be fast to load the models from disk.
+
+One way to make this process about 10x faster is to convert the models to pytorch format using the script `convert-to-torch.py`. Create a folder called `torch-dumps` and then make the conversion with:
+
+    python convert-to-torch.py models/model-name/
+
+The output model will be saved to `torch-dumps/model-name.pt`. This is the default way to load all models except for `gpt-neox-20b`, `opt-13b`, `OPT-13B-Erebus`, `gpt-j-6B`, and `flan-t5`. I don't remember why these models are exceptions.
+
+If I get enough ⭐s on this repository, I will make the process of loading models more transparent and straightforward.
+
 ## Starting the webui
 
     conda activate textgen

From e0583f0ec22150314dfed547f30c8b4f0bfa8ee5 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 21 Dec 2022 16:52:23 -0300
Subject: [PATCH 2/3] Update README.md

---
 README.md | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index f8029960..75b1633b 100644
--- a/README.md
+++ b/README.md
@@ -39,7 +39,7 @@ One way to make this process about 10x faster is to convert the models to pytorc
 
 The output model will be saved to `torch-dumps/model-name.pt`. This is the default way to load all models except for `gpt-neox-20b`, `opt-13b`, `OPT-13B-Erebus`, `gpt-j-6B`, and `flan-t5`. I don't remember why these models are exceptions.
 
-If I get enough ⭐s on this repository, I will make the process of loading models more transparent and straightforward.
+If I get enough ⭐s on this repository, I will make the process of loading models saner and more customizable.
 
 ## Starting the webui
 
@@ -47,3 +47,7 @@ If I get enough ⭐s on this repository, I will make the process of loading mode
     python server.py
 
 Then browse to `http://localhost:7860/?__theme=dark`
+
+## Contributing
+
+Pull requests are welcome.

From a90097e0b8be13f68d6c9896591d38c29f01afa7 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 21 Dec 2022 21:49:03 -0300
Subject: [PATCH 3/3] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 75b1633b..5c531e0f 100644
--- a/README.md
+++ b/README.md
@@ -10,7 +10,7 @@ Its goal is to become the [AUTOMATIC1111/stable-diffusion-webui](https://github.
 
     conda env create -f environment.yml
 
-This installs the CUDA version of pytorch, which assumes that you have a NVIDIA GPU. If you want to run this on an AMD GPU, you should install the ROCM version of pytorch instead.
+This installs the CUDA version of pytorch, which assumes that you have a NVIDIA GPU. If you want to run this on an AMD GPU, you should install the ROCm version of pytorch instead.
 
 ## Downloading models