Skip to content

Commit

Permalink
Merge branch 'main' into kaiembed
Browse files Browse the repository at this point in the history
  • Loading branch information
db0 committed Sep 16, 2023
2 parents 092dd75 + 6223435 commit 4b18948
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 6 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ This repo contains the latest implementation for the [AI Horde](https://aihorde.

Alternatively you can become an Alchemist worker which is much more lightweight and can even run on CPU (i.e. without a GPU).

Please note that **AMD card are** not currently supported*, but may be in the future.
Please note that **AMD card are not currently supported**, but may be in the future.

To run the bridge, simply follow the instructions for your own OS

Expand Down Expand Up @@ -70,7 +70,7 @@ This script can take 10-15 minutes to complete.

In order to connect to the horde with your username and a good worker name, you need to configure your horde bridge. To this end, we've developed an easy WebUI you can use

To load it, simply run `bridge-webui`. It will then show you a URL you can open with your browser. Open it and it will allow you to tweak all horde options. Once you press `Update Bridge` it will create a `bridgeData.yaml` file with all the options you set.
To load it, simply run `bridge-webui`. It will then show you a URL you can open with your browser. Open it and it will allow you to tweak all horde options. Once you press `Save Configuration` it will create a `bridgeData.yaml` file with all the options you set.

Fill in at least:
* Your worker name (has to be unique horde-wide)
Expand Down
1 change: 1 addition & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
--extra-index-url https://download.pytorch.org/whl/cu118
torch # We have to place it here otherwise it installs the CPU version
Pillow>=10.0.1
pydantic==1.10.12
horde_model_reference~=0.2.2
transformers==4.30.2
Expand Down
6 changes: 3 additions & 3 deletions show_available_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,9 @@
mm = SharedModelManager.manager

filtered_models = mm.compvis.get_filtered_models(type="ckpt")

sorted_model_names = sorted(filtered_models.keys())

ppmodels = ""
for model_name in sorted_model_names:
if model_name == "LDSR":
Expand All @@ -29,4 +29,4 @@
ppmodels += f" : {filtered_models[model_name].get('description')}"
ppmodels += "\n"
print(f"## Known ckpt Models ##\n{ppmodels}")
input("Press ENTER to continue")
input("Press ENTER to continue")
2 changes: 1 addition & 1 deletion worker/consts.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
BRIDGE_VERSION = 24
RELEASE = f"{BRIDGE_VERSION}.0.0"
RELEASE = f"{BRIDGE_VERSION}.0.2"
BRIDGE_CONFIG_FILE = "bridgeData.yaml"
KNOWN_UPSCALERS = {
"RealESRGAN_x4plus",
Expand Down

0 comments on commit 4b18948

Please sign in to comment.