ModuleNotFoundError：没有名为“llama”的模块

Question

我尝试执行以下代码

import itertools
import jsonlines

from datasets import load_dataset
from pprint import pprint

from llama import BasicModelRunner
from transformers import AutoTokenizer, AutoModelForCausalLM
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

出现以下错误

Traceback (most recent call last):
  File "/Users/jigarpatel/LLMs/Course/FineTuningLLM101/FineTuningLLM101/.venv/bin/instruction_tuning_lab_student.py", line 7, in <module>
    from llama import BasicModelRunner
ModuleNotFoundError: No module named 'llama'

请帮忙，提前致谢。

我正在本地主机上的 PyCharm Community Edition 上运行代码。

Answer 1

试试这个：

pip install lamini

然后：

from llama import BasicModelRunner