Comments (2)
我在windows下构建也遇到了,把Dockerfile.base中这段代码
RUN cat > /get-models.py <<EOF
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
AutoModelForSeq2SeqLM.from_pretrained('Helsinki-NLP/opus-mt-zh-en')
AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-zh-en')
pipeline('text-generation', model='succinctly/text2image-prompt-generator')
EOF
改为
RUN echo "from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline" > /get-models.py && \
echo "AutoModelForSeq2SeqLM.from_pretrained('Helsinki-NLP/opus-mt-zh-en')" >> /get-models.py && \
echo "AutoTokenizer.from_pretrained('Helsinki-NLP/opus-mt-zh-en')" >> /get-models.py && \
echo "pipeline('text-generation', model='succinctly/text2image-prompt-generator')" >> /get-models.py
就可以了
from docker-prompt-generator.
构建Dockerfile.gpu出错时,将下列代码进行替换:
RUN cat > /get-models.py <<EOF from clip_interrogator import Config, Interrogator import torch config = Config() config.device = 'cuda' if torch.cuda.is_available() else 'cpu' config.blip_offload = False if torch.cuda.is_available() else True config.chunk_size = 2048 config.flavor_intermediate_count = 512 config.blip_num_beams = 64 config.clip_model_name = "ViT-H-14/laion2b_s32b_b79k" ci = Interrogator(config) EOF
->
RUN echo "from clip_interrogator import Config, Interrogator" >> /get-models.py && \ echo "import torch" >> /get-models.py && \ echo "config = Config()" >> /get-models.py && \ echo "config.device = 'cuda' if torch.cuda.is_available() else 'cpu'" >> /get-models.py && \ echo "config.blip_offload = False if torch.cuda.is_available() else True" >> /get-models.py && \ echo "config.chunk_size = 2048" >> /get-models.py && \ echo "config.flavor_intermediate_count = 512" >> /get-models.py && \ echo "config.blip_num_beams = 64" >> get-models.py && \ echo "config.clip_model_name = \"ViT-H-14/laion2b_s32b_b79k\"" >> /get-models.py && \ echo "ci = Interrogator(config)" >> /get-models.py
from docker-prompt-generator.
Related Issues (20)
- I ask a question ?: do the prompt continue is using imagine with previous image url ? HOT 3
- pull access denied for soulteary/prompt-generator, repository does not exist or may require 'docker login': denied: requested access to the resource is denied HOT 1
- 运行 GPU 镜像时出错
- 我这边老是报这个错误
- PLAN: let build or deploy easier
- feat: parse image from URL
- 401 Unauthorized
- huggingface 被墙,nvcr.io mac os 挂代理会导致无法下载pytorch,难道只能用国外服务器吗? HOT 2
- ubantu,GPU模式,容器啓動時報錯 HOT 1
- 部署好之后执行报错 HOT 4
- ERROR: failed to solve authorization failed
- 老是这个问题,gradio更新或安装2.4.5也没用
- 从文本中生成时报错 HOT 1
- docker build image failed HOT 1
- 部署报错:failed to solve: rpc error: code = Unknown desc HOT 1
- 看好你老哥,但是非得docker才能安装吗 HOT 1
- docker build都会报错,手动能解决,但是你的get-models.py文件内容是什么呢? HOT 1
- building base image error HOT 1
- 部署报错 HOT 5
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from docker-prompt-generator.