CTranslate2
CTranslate2 是一個 C++ 和 Python 函式庫，用於 Transformer 模型的高效率推論。
此專案實作了一個自訂執行時間，應用許多效能最佳化技術，例如權重量化、層融合、批次重新排序等，以加速並減少 Transformer 模型在 CPU 和 GPU 上的記憶體使用量。
完整的功能和支援模型清單包含在專案的儲存庫中。若要開始，請查看官方的快速入門指南。
若要使用，您應該已安裝 ctranslate2 python 套件。
%pip install --upgrade --quiet ctranslate2
若要將 Hugging Face 模型與 CTranslate2 搭配使用，必須先使用 ct2-transformers-converter 命令將其轉換為 CTranslate2 格式。此命令採用預訓練模型名稱和已轉換模型目錄的路徑。
# conversation can take several minutes
!ct2-transformers-converter --model meta-llama/Llama-2-7b-hf --quantization bfloat16 --output_dir ./llama-2-7b-ct2 --force
Loading checkpoint shards: 100%|██████████████████| 2/2 [00:01<00:00, 1.81it/s]
from langchain_community.llms import CTranslate2
llm = CTranslate2(
# output_dir from above:
model_path="./llama-2-7b-ct2",
tokenizer_name="meta-llama/Llama-2-7b-hf",
device="cuda",
# device_index can be either single int or list or ints,
# indicating the ids of GPUs to use for inference:
device_index=[0, 1],
compute_type="bfloat16",
)
API 參考：CTranslate2
單次呼叫
print(
llm.invoke(
"He presented me with plausible evidence for the existence of unicorns: ",
max_length=256,
sampling_topk=50,
sampling_temperature=0.2,
repetition_penalty=2,
cache_static_prompt=False,
)
)
He presented me with plausible evidence for the existence of unicorns: 1) they are mentioned in ancient texts; and, more importantly to him (and not so much as a matter that would convince most people), he had seen one.
I was skeptical but I didn't want my friend upset by his belief being dismissed outright without any consideration or argument on its behalf whatsoever - which is why we were having this conversation at all! So instead asked if there might be some other explanation besides "unicorning"... maybe it could have been an ostrich? Or perhaps just another horse-like animal like zebras do exist afterall even though no humans alive today has ever witnesses them firsthand either due lacking accessibility/availability etc.. But then again those animals aren’ t exactly known around here anyway…” And thus began our discussion about whether these creatures actually existed anywhere else outside Earth itself where only few scientists ventured before us nowadays because technology allows exploration beyond borders once thought impossible centuries ago when travel meant walking everywhere yourself until reaching destination point A->B via footsteps alone unless someone helped guide along way through woods full darkness nighttime hours
多次呼叫：
print(
llm.generate(
["The list of top romantic songs:\n1.", "The list of top rap songs:\n1."],
max_length=128,
)
)
generations=[[Generation(text='The list of top romantic songs:\n1. “I Will Always Love You” by Whitney Houston\n2. “Can’t Help Falling in Love” by Elvis Presley\n3. “Unchained Melody” by The Righteous Brothers\n4. “I Will Always Love You” by Dolly Parton\n5. “I Will Always Love You” by Whitney Houston\n6. “I Will Always Love You” by Dolly Parton\n7. “I Will Always Love You” by The Beatles\n8. “I Will Always Love You” by The Rol', generation_info=None)], [Generation(text='The list of top rap songs:\n1. “God’s Plan” by Drake\n2. “Rockstar” by Post Malone\n3. “Bad and Boujee” by Migos\n4. “Humble” by Kendrick Lamar\n5. “Bodak Yellow” by Cardi B\n6. “I’m the One” by DJ Khaled\n7. “Motorsport” by Migos\n8. “No Limit” by G-Eazy\n9. “Bounce Back” by Big Sean\n10. “', generation_info=None)]] llm_output=None run=[RunInfo(run_id=UUID('628e0491-a310-4d12-81db-6f2c5309d5c2')), RunInfo(run_id=UUID('f88fdbcd-c1f6-4f13-b575-810b80ecbaaf'))]
將模型整合到 LLMChain 中
from langchain.chains import LLMChain
from langchain_core.prompts import PromptTemplate
template = """{question}
Let's think step by step. """
prompt = PromptTemplate.from_template(template)
llm_chain = LLMChain(prompt=prompt, llm=llm)
question = "Who was the US president in the year the first Pokemon game was released?"
print(llm_chain.run(question))
API 參考：LLMChain | PromptTemplate
Who was the US president in the year the first Pokemon game was released?
Let's think step by step. 1996 was the year the first Pokemon game was released.
\begin{blockquote}
\begin{itemize}
\item 1996 was the year Bill Clinton was president.
\item 1996 was the year the first Pokemon game was released.
\item 1996 was the year the first Pokemon game was released.
\end{itemize}
\end{blockquote}
I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.
Comment: @JoeZ. I'm not sure if this is a valid question, but I'm sure it's a fun one.