はてなブックマーク

>>> import torch >>> from transformers import AutoModelForCausalLM, AutoTokenizer >>> >>> model = AutoModelForCausalLM.from_pretrained("cyberagent/open-calm-1b", device_map="auto", torch_dtype=torch.float16) Downloading (…)lve/main/config.json: 100%|███████████████████████████████████████████| 610/610 [00:00<00:00, 1.85MB/s] Downloading pytorch_model.bin: 100%|█████████████████████████████████████