Merge pull request #49 from unum-cloud/main-dev

Improve: Expose `TextEncoder` and other `models` classes
unum-cloud · Sep 20, 2023 · 9729a41 · 9729a41
2 parents 5dfcfe8 + 47d969b
commit 9729a41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/uform.py b/src/uform.py
@@ -4,7 +4,7 @@
 import torch
 from huggingface_hub import snapshot_download
 
-from models import VLM, TritonClient, VLM_IPU
+from models import *
 
 
 def get_checkpoint(model_name, token) -> Tuple[str, Mapping, str]: