[issue]: AttributeError: module 'jtorch.cuda' has no attribute 'amp' after running collection = chroma_client.create_collection(name="my_collection")

[BasicAILinDu]

What happened?

from .adam import Adam File "/opt/homebrew/Caskroom/miniforge/base/lib/python3.10/site-packages/torch/optim/adam.py", line 35, in grad_scaler: Optional[torch.cuda.amp.GradScaler] = None, AttributeError: module 'jtorch.cuda' has no attribute 'amp'

Versions

chroma 0.3.21 macos

Relevant log output

No response