more archs added

norsulabs · Nov 24, 2024 · 143d0c8 · 143d0c8
1 parent 5fbda1b
commit 143d0c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/infero/pull/supported_architectures.yaml b/infero/pull/supported_architectures.yaml
@@ -2,4 +2,6 @@ architectures:
   - name: 'RobertaForSequenceClassification'
   - name: 'BertForSequenceClassification'
   - name: 'DistilBertForSequenceClassification'
-  - name: 'AlbertForSequenceClassification'
+  - name: 'AlbertForSequenceClassification'
+  - name: 'XLMRobertaForSequenceClassification'
+  - name: 'XLMForSequenceClassification'
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "infero"
-version = "0.0.11"
+version = "0.0.12"
 description = "Easily download, convert, and host your models using the ONNX runtime"
 authors = ["ameen-91 <[email protected]>"]
 license = "MIT"