- 博客(3)
- 收藏
- 关注
RSS订阅原创 解决使用预训练模型时无法从Huggingface下载权重文件的问题
在使用transformers库中的一些类时（如DistilBertModel，DistilBertTokenizer，AutoTokenizer）时，会自动从huggingface.co上下载对应模型的权重文件以及一些配置文件，如：vocab.txt，config.json。最近在搞condition generation的项目，主要是参考Stable Diffusion的模型架构，按照甲方需求定制文生图模型。把原来的huggingface.co注释掉，改成国内镜像地址：hf-mirror.com。
2024-12-16 20:58:51
266
原创 clip_interrogator库
然后想着用clip_interrogator库在cifar10-64数据集先试一下。建议对于stable diffusion 1.X版本，clip model使用ViT-L-14/openai；对于stable diffusion 2.0版本，clip model使用ViT-H-14/laion2b_s32b_b79k。在clip-interrogator==0.6.0的环境下两种clip_model均不会出现报错。我自己的环境是miniconda，这点没啥区别。
2024-12-16 20:16:48
323
原创 Python函数
torch.gather()函数用于根据指定索引从出入的张量中收集元素。index：一个包含了从input中收集元素的索引的。dim：指定收集操作沿着哪个维度进行。out：可选参数，指定输出张量的位置。input：输入张量。
2023-11-23 18:13:45
672
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人