owiewowe https://blog.hatena.ne.jp/owiewowe/ 19日に更新してた https://owiewowe.hatenablog.com/ win11home、i7、32G、GTX1060 6G 環境です。cyberagent/open-calm-1b をいじっていて cyberagent/open-calm-3b ならおそらく device_map をいじってやったら動くのではないかなとやってみたら動かせたのでせっかくなので書いておこうかと。*1 import torch from transformers import AutoModelForCausalLM, AutoTokenizer from accelerate import infer_auto_device_map model_name="cyberagent/o… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Fowiewowe.hatenablog.com%2Fentry%2F2023%2F05%2F19%2F000000_3" title="GTX1060 6G にて device_map を少しいじって cyberagent/open-calm-3b を動かしてみた。 - 19日に更新してた" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> Hatena Blog https://hatena.blog 2023-05-19 00:00:00 rich https://owiewowe.hatenablog.com/entry/2023/05/19/000000_3 1.0 100%