machineuser commited on
Commit
27369a0
1 Parent(s): 28fb656

Sync widgets demo

Browse files
packages/tasks/src/library-ui-elements.ts CHANGED
@@ -35,11 +35,11 @@ function nameWithoutNamespace(modelId: string): string {
35
 
36
  //#region snippets
37
 
38
- const adapter_transformers = (model: ModelData) => [
39
- `from transformers import ${model.config?.adapter_transformers?.model_class}
40
 
41
- model = ${model.config?.adapter_transformers?.model_class}.from_pretrained("${model.config?.adapter_transformers?.model_name}")
42
- model.load_adapter("${model.id}", source="hf")`,
43
  ];
44
 
45
  const allennlpUnknown = (model: ModelData) => [
@@ -576,11 +576,11 @@ model = AutoModel.load_from_hf_hub("${model.id}")`,
576
 
577
  export const MODEL_LIBRARIES_UI_ELEMENTS: Partial<Record<ModelLibraryKey, LibraryUiElement>> = {
578
  "adapter-transformers": {
579
- btnLabel: "Adapter Transformers",
580
- repoName: "adapter-transformers",
581
- repoUrl: "https://github.com/Adapter-Hub/adapter-transformers",
582
- docsUrl: "https://huggingface.co/docs/hub/adapter-transformers",
583
- snippets: adapter_transformers,
584
  },
585
  allennlp: {
586
  btnLabel: "AllenNLP",
 
35
 
36
  //#region snippets
37
 
38
+ const adapters = (model: ModelData) => [
39
+ `from adapters import AutoAdapterModel
40
 
41
+ model = AutoAdapterModel.from_pretrained("${model.config?.adapter_transformers?.model_name}")
42
+ model.load_adapter("${model.id}", set_active=True)`,
43
  ];
44
 
45
  const allennlpUnknown = (model: ModelData) => [
 
576
 
577
  export const MODEL_LIBRARIES_UI_ELEMENTS: Partial<Record<ModelLibraryKey, LibraryUiElement>> = {
578
  "adapter-transformers": {
579
+ btnLabel: "Adapters",
580
+ repoName: "adapters",
581
+ repoUrl: "https://github.com/Adapter-Hub/adapters",
582
+ docsUrl: "https://huggingface.co/docs/hub/adapters",
583
+ snippets: adapters,
584
  },
585
  allennlp: {
586
  btnLabel: "AllenNLP",
packages/tasks/src/model-libraries.ts CHANGED
@@ -5,7 +5,7 @@
5
  * File formats live in an enum inside the internal codebase.
6
  */
7
  export enum ModelLibrary {
8
- "adapter-transformers" = "Adapter Transformers",
9
  "allennlp" = "allenNLP",
10
  "asteroid" = "Asteroid",
11
  "bertopic" = "BERTopic",
 
5
  * File formats live in an enum inside the internal codebase.
6
  */
7
  export enum ModelLibrary {
8
+ "adapter-transformers" = "Adapters",
9
  "allennlp" = "allenNLP",
10
  "asteroid" = "Asteroid",
11
  "bertopic" = "BERTopic",