liwenyun
/
CLIP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add info on how to run tests to README.md

This commit is contained in:
bartosz roguski 2025-02-12 14:06:40 +01:00
parent af0f80c1e6
commit 6d3570d0e2
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -221,6 +221,12 @@ docker run -it --runtime=habana clip_hpu:latest
You do not need to change the code to leverage Intel® Gaudi® HPU. The `get_device_initial()` function will automatically detect the correct device and return the appropriate device name. So no changes are required. You do not need to change the code to leverage Intel® Gaudi® HPU. The `get_device_initial()` function will automatically detect the correct device and return the appropriate device name. So no changes are required.
### Run the Tests
```bash
pytest
```
This will run the tests and verify that the model is working correctly.
## See Also ## See Also

1
tests/test_consistency.py
View File

@ -2,6 +2,7 @@ import numpy as np
import pytest import pytest
import torch import torch
from PIL import Image from PIL import Image
import habana_frameworks.torch
import clip import clip