[Tracking] Model loading/caching enhancements

[CharlieFRuan]

[Tracking] Model loading/caching enhancements

[tqchen]

Sorry for chiming in late. For the caching layer, ideally we would like something that comes with minimal dependency. Spefically, we should:

• Make ArtifactCache https://github.com/apache/tvm/blob/main/web/src/runtime.ts#L991 an interface ArtifactCache in a new file artifact_cache.ts
• Provide implementatons ArtifactCacheIndexDB, and ArtifactCacheBasic(via caches)
• Provide parallel download methods in the same file
• Optionally, allow injection of additional class of ArtifactCache that can be implemented via other means.

This way the default implementation won't come with extra dependency via IndexDB API.

[DiegoCao]

Hi David, thanks for offering the help! I'm looking into the TVMjs as well and we need to make changes in module TVMjs.

[DavidGOrtega]

@CharlieFRuan I think all of them are in TVMjs. Parallelise downloads and change the cache layer for something much more agnostic. As I say I would use level because we can then me it work with different caches. Its a facade

[DiegoCao]

@DavidGOrtega I can work on C2 first and work on C4 after you built the indexDB and levels.

[DavidGOrtega]

Perfect so I do a PR for C1 and another for C3

[DavidGOrtega]

Great! I was thinking to group all of them too. I would implement a facade and then we can implement specific storages.
I suggest to use level if we are able to just use a package.

I can definitely take this one! Im actually taking it for a project of mine.
My question would be, Are we attacking TVMjs or web-llm? Right now the cache mechanism is the one implemented in TVMjs. And I know for sure that mlc-ai is actually related to TVM but Im afraid of how long does it take to deploy something in TVM. Currently they have 220 PRs some of them from 2021 (including yours) @CharlieFRuan

[CharlieFRuan]

I can definitely take this one! Im actually taking it for a project of mine.

@DavidGOrtega Thanks for offering help! You are referring to item C3 right?

My question would be, Are we attacking TVMjs or web-llm?

I think we should make the changes in TVMjs. I wouldn't be too worried about getting things merged there:)

Also cc @DiegoCao who is looking into IndexedDB as well.

[DavidGOrtega]

let me know what works for you @DiegoCao. What do you want to pick?

[CharlieFRuan]

I think we can go with your suggestion and use level here. Looking forward to the change!

[DiegoCao]

Sorry for chiming in late. For the caching layer, ideally we would like something that comes with minimal dependency. Spefically, we should:

• Make ArtifactCache https://github.com/apache/tvm/blob/main/web/src/runtime.ts#L991 an interface ArtifactCache in a new file artifact_cache.ts
• Provide implementatons ArtifactCacheIndexDB, and ArtifactCacheBasic(via caches)
• Provide parallel download methods in the same file
• Optionally, allow injection of additional class of ArtifactCache that can be implemented via other means.

This way the default implementation won't come with extra dependency via IndexDB API.

Received, will do a PR for C0 for migrating old ArtifactCache to interface and implementation of the existing Basic Approach.

[bakkot]

Another nice feature, especially if you're doing C4, would be to allow inspecting items in the cache "downloading" them, i.e. copying them out of the cache to disk (so that users don't have to find them wherever their browser is storing files).