小编Moh*_*ghi的帖子

导入错误:使用 `low_cpu_mem_usage=True` 或 `device_map` 需要加速:`pip install Accelerate`

我尝试使用 HuggingFace Transformer 通过运行以下代码来创建管道(代码在 SageMaker Jupyter 实验室上运行):

pipeline = transformers.pipeline(
    "text-generation",
    model="meta-llama/Llama-2-7b-chat-hf",
    torch_dtype=torch.float16,
    device_map="auto",
)
Run Code Online (Sandbox Code Playgroud)

但是,它会生成以下错误:

ImportError: Using `low_cpu_mem_usage=True` or a `device_map` requires Accelerate: `pip install accelerate`
Run Code Online (Sandbox Code Playgroud)

参考:

Packages version:print(accelerate.__version__), print(transformers.__version__)
0.21.0
4.31.0
Run Code Online (Sandbox Code Playgroud)

accelerate huggingface-transformers amazon-sagemaker-studio

12
推荐指数
1
解决办法
2万
查看次数

在 Pandas DataFrame 中将字符串列直接转换为日期格式(不是日期时间)

我有一个以下的 Pandas DataFrame:

df = pd.DataFrame({'a': ['2020-01-02', '2020-01-02']})

显然,列“a”是字符串。我想将其转换为“日期”类型;这就是我所做的:

df['a'] = df['a'].apply(pd.to_datetime).dt.date

它有效,但实际上我的 DataFrame 有 500,000 多行。看起来效率很低。有没有办法直接更有效地将字符串列转换为日期列?

python dataframe pandas

6
推荐指数
1
解决办法
1万
查看次数