Comments (5)
echo840
commented on July 1, 2024
CUDA_VISIBLE_DEVICES=4 python demo_textmonkey.py -c model_path
Maybe you can add "CUDA_VISIBLE_DEVICES=4".
from monkey.
echo840
commented on July 1, 2024
device_map="cuda:4"
input_ids=input_ids.to("cuda:4"),
attention_mask=attention_mask.to("cuda:4"),
Alternatively, you can modify the code here.
from monkey.
Mararesliu
commented on July 1, 2024
The first way not change too.
The second way demo can no start ,rasie error
from monkey.
echo840
commented on July 1, 2024
Is your GPU functioning properly? Can you run other models on "cuda:4"? I don't think this is an issue with our code.
from monkey.
Mararesliu
commented on July 1, 2024
It works, many thinks.
from monkey.
Related Issues (20)
- TextMonkey问题 HOT 1
- A100 40G可以跑通训练吗？全参数SFT和LoRA我在A100 40G报OOM，我debug看到是self.visual.encode(images)就报OOM了 HOT 14
- Data Access HOT 2
- TextMonkey RuntimeError HOT 8
- 为什么文档理解的输入不是pdf或者doc文档，而是图片？ HOT 1
- textmonkey支持多图输入吗 HOT 1
- Will Rico data be released? HOT 4
- How to finetune only one subnetwork using Deepspeed + Transformers
- How to finetune certain params via from HF's transformers, a
- vizwiz的准确率仅有37.62？表中的结果为61.2?QwenVL是35.2，请问是数据填写错误吗？ HOT 8
- Get the embeddings of the image. HOT 1
- Textmonkey有推理代码吗，为什么web demo运行起来不回答 HOT 1
- demo doesn't give OCR with grounding HOT 3
- 模型加载问题 HOT 8
- What is the various task-specific augmentations to different dataset mentioned in TextMonkey Sec3.5 ? HOT 2
- eval/eval_doc.sh中运行是的哪个py脚本呢？ HOT 2
- demo_textmonkey.py 模型加载问题 HOT 2
- RunTimeError: Numpy is not available
- Token sampler代码与论文的差异
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 monkey.