You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
imaginAIry/tests
Bryce 9614a82f12 fix: use float32 at image render time for wider cpu compatibility
If the x_sample was a bfloat on the gpu but the cpu doesn't support bfloat, that can cause a TypeError
```
  File "/home/stdiff/.local/lib/python3.10/site-packages/imaginairy/api.py", line 292, in imagine
    x_sample.cpu().numpy(), "c h w -> h w c"

TypeError: Got unsupported ScalarType BFloat16`
```
2 years ago
..
data
Dockerfile tests: add docker image for testing environment. minor test improvements 2 years ago
__init__.py
conftest.py tests: add docker image for testing environment. minor test improvements 2 years ago
test_clip_embedder.py tests: add docker image for testing environment. minor test improvements 2 years ago
test_cmds.py tests: add docker image for testing environment. minor test improvements 2 years ago
test_enhancers.py tests: add docker image for testing environment. minor test improvements 2 years ago
test_experiments.py
test_guidance.py
test_imagine.py fix: use float32 at image render time for wider cpu compatibility 2 years ago
test_safety.py tests: add docker image for testing environment. minor test improvements 2 years ago
test_schema.py