hegelai / prompttools

Open-source tools for prompt testing and experimentation, with support for both LLMs (e.g. OpenAI, LLaMA) and vector databases (e.g. Chroma, Weaviate, LanceDB).
http://prompttools.readthedocs.io
Apache License 2.0
2.55k stars 216 forks source link

Remove `torch` reference #85

Closed NivekT closed 10 months ago

NivekT commented 10 months ago

Hey @HashemAlsaket,

I don't see any direct usage of torch within the experiment. I am wondering if this is safe to remove?

I understand that other library might have dependencies on torch such that it might get installed anyway, but I'm wondering if we need to import it here.

Let me know! Thanks for your work!