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

d577c26

1 Parent(s): ffa3a23

Training in progress, step 160

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:af04f516604e06958c79ad27fb8aaf8e7a3262ffc8299a5c6416a831f6a24eb1
 size 522284877

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

last-checkpoint/adapter_model.bin CHANGED Viewed

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

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

last-checkpoint/optimizer.pt CHANGED Viewed

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

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

last-checkpoint/rng_state.pth CHANGED Viewed

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

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

last-checkpoint/scheduler.pt CHANGED Viewed

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

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

last-checkpoint/trainer_state.json CHANGED Viewed

@@ -1,8 +1,8 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 37.5,
-  "global_step": 150,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -96,11 +96,17 @@
       "learning_rate": 0.0002,
       "loss": 0.0091,
       "step": 150
     }
   ],
   "max_steps": 200,
   "num_train_epochs": 50,
-  "total_flos": 4.904852547895296e+16,
   "trial_name": null,
   "trial_params": null
 }

 {
   "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,
       "learning_rate": 0.0002,
       "loss": 0.0091,
       "step": 150
+    },
+    {
+      "epoch": 40.0,
+      "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
 }