Instructions to use subset-data/falcon-7b-bt with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use subset-data/falcon-7b-bt with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="subset-data/falcon-7b-bt", trust_remote_code=True)

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("subset-data/falcon-7b-bt", trust_remote_code=True, device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use subset-data/falcon-7b-bt with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "subset-data/falcon-7b-bt"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "subset-data/falcon-7b-bt",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/subset-data/falcon-7b-bt

SGLang

How to use subset-data/falcon-7b-bt with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "subset-data/falcon-7b-bt" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "subset-data/falcon-7b-bt",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "subset-data/falcon-7b-bt" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "subset-data/falcon-7b-bt",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use subset-data/falcon-7b-bt with Docker Model Runner:
```
docker model run hf.co/subset-data/falcon-7b-bt
```

brian-todd commited on Jul 20, 2023

Commit

32702df

1 Parent(s): d577c26

Training in progress, step 170

Browse files

Files changed (6) hide show

adapter_model.bin +1 -1
last-checkpoint/adapter_model.bin +1 -1
last-checkpoint/optimizer.pt +1 -1
last-checkpoint/rng_state.pth +1 -1
last-checkpoint/scheduler.pt +1 -1
last-checkpoint/trainer_state.json +9 -3

adapter_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:4b731b33a663df9b5d325ea24811ab15b363e54445aff4ec4bd6fe507e6467d7
 size 522284877

last-checkpoint/adapter_model.bin CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:4b731b33a663df9b5d325ea24811ab15b363e54445aff4ec4bd6fe507e6467d7
 size 522284877

last-checkpoint/optimizer.pt CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:46a23fcd6874a1853674994bee58a2a786096d6a41cbad082a14e2a21191dc30
 size 1044539653

 version https://git-lfs.github.com/spec/v1
+oid sha256:51b0f686c1f2bee0e68037a4cca40921cdb05f7528cabced5713e8ae0cb43297
 size 1044539653

last-checkpoint/rng_state.pth CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:aeb77c5432001f75ca5c8c8d2df598ea67c727a71f0a1359228917568a1d0916
 size 14575

last-checkpoint/scheduler.pt CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4e87ab8212c9c1b5c9ac2af513be72fc40777cbdb25e7df360e56de5cd75aab0
 size 627

 version https://git-lfs.github.com/spec/v1
+oid sha256:d3841b7d3094f97af299f97dddaa8fe1e5f1142ed0601dee132ea7f776f6ffff
 size 627

last-checkpoint/trainer_state.json CHANGED Viewed

@@ -1,8 +1,8 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 40.0,
-  "global_step": 160,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -102,11 +102,17 @@
       "learning_rate": 0.0002,
       "loss": 0.009,
       "step": 160
     }
   ],
   "max_steps": 200,
   "num_train_epochs": 50,
-  "total_flos": 5.23073568374784e+16,
   "trial_name": null,
   "trial_params": null
 }

 {
   "best_metric": null,
   "best_model_checkpoint": null,
+  "epoch": 42.5,
+  "global_step": 170,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
       "learning_rate": 0.0002,
       "loss": 0.009,
       "step": 160
+    },
+    {
+      "epoch": 42.5,
+      "learning_rate": 0.0002,
+      "loss": 0.0088,
+      "step": 170
     }
   ],
   "max_steps": 200,
   "num_train_epochs": 50,
+  "total_flos": 5.558694508363776e+16,
   "trial_name": null,
   "trial_params": null
 }