Views
No views yet
pip install hf_hub_ctranslate2>=1.0.0 ctranslate2>=3.13.0compute_type=int8_float16 for device="cuda"compute_type=int8 for device="cpu"1from hf_hub_ctranslate2 import TranslatorCT2fromHfHub, GeneratorCT2fromHfHub
2
3model_name = "michaelfeil/ct2fast-flan-alpaca-base"
4model = TranslatorCT2fromHfHub(
5 # load in int8 on CUDA
6 model_name_or_path=model_name,
7 device="cuda",
8 compute_type="int8_float16"
9)
10outputs = model.generate(
11 text=["How do you call a fast Flan-ingo?", "Translate to german: How are you doing?"],
12 min_decoding_length=24,
13 max_decoding_length=32,
14 max_input_length=512,
15 beam_size=5
16)
17print(outputs)