-
Notifications
You must be signed in to change notification settings - Fork 2.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: add nebullvm as backend #697
base: main
Are you sure you want to change the base?
Conversation
awsome job. The current PR cannot pass CI checking. Please follow this guidelines https://github.com/jina-ai/jina/blob/master/CONTRIBUTING.md |
BTW., please also use
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please check my comments 🍻
hi @morgoth95 is it possible to open PR modification permission to us? we could take care of it on the styling and minors things. Reference here: https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/allowing-changes-to-a-pull-request-branch-created-from-a-fork |
Hello @hanxiao! Actually, I already activated the option "Allow edits by maintainers" when I created the PR 😄 In theory all of you with writing access to clip-as-a-service should be able to directly push new commits to my branch 😉 |
README.md
Outdated
@@ -45,13 +45,20 @@ curl -X POST http://demo-cas.jina.ai:51001/post -H 'Content-Type: application/js | |||
## Install | |||
|
|||
CLIP-as-service consists of two Python packages `clip-server` and `clip-client` that can be installed _independently_. Both require Python 3.7+. | |||
`nebullvm` installation will need on some platforms python |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This sentence is more about the setup. It's better to be placed in the following Install server
section.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed in 01f7f08
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
And please add unit test for nebullvm
docs/user-guides/server.md
Outdated
|
||
python -m clip_server nebullvm_flow.yml |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should be nebullvm-flow.yml
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
solved in f06133e
|
||
def __enter__(self): | ||
if self.device == "cpu" and torch.cuda.is_available(): | ||
self.cuda_str = os.environ.get("CUDA_VISIBLE_DEVICES") or "1" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this a bug? maybe we can use self.cuda_str = os.environ.get('CUDA_VISIBLE_DEVICES', None)
, otherwise the CUDA_VISIBLE_DEVICES
will be updated to "1"
at exiting.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Solved in f9ce0ae
@morgoth95 how is going now? Just wondering do we have the chance to merge this PR this week? |
Sorry for the delay, we were working on the latest release of nebullvm! I should have fixed the issues you pointed out! I think we can try to accelerate the process and merge the PR this week. |
@morgoth95 Congrats. |
c0808e0
to
b42d031
Compare
FYI, this PR has one ore more bad commit messages,
|
it should have been fixed now! Let me know if further changes are needed |
@morgoth95 Please rebase this PR so that it does not show unwanted commits |
@morgoth95 please remember to install pre-commit to format your code before commits:
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please format your code via:
$ pre-commit run --all
Solved in 932cfe6 |
Codecov Report
@@ Coverage Diff @@
## main #697 +/- ##
===========================================
- Coverage 81.58% 47.87% -33.72%
===========================================
Files 21 18 -3
Lines 1575 1201 -374
===========================================
- Hits 1285 575 -710
- Misses 290 626 +336
Flags with carried forward coverage won't be shown. Click here to find out more.
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
@morgoth95 The unittest seems broken |
@morgoth95 Hi, I hope you are well. Did you get a chance to continue this PR to pass the unit test? |
Hi Felix, I'm working on the tests. A couple of days and I'll close the open tasks. Looking forward to the integration! |
@numb3r3 Hi, I hope all is fine on the Jina side. I took a look at the failed tests and I may have identified the source of the problem. Unfortunately it will require some work on our end, but we think it will be fixed with the next nebullvm release (it will happen in about a week ;)). I will get back to you as soon as the problem is fixed :) |
@morgoth95 very nice, looking forward the next nebullvm release. |
Add nebullvm as a model backend in a similar way to ONNXruntime backend