Fix trust_remote_code in snippets (#467)

At the moment, we only add `trust_remote_code` in the cases where
there's a custom class, but this will fail when there are other custom
components, such as tokenizer/config/etc. Instead, we just use
`trust_remote_code` if there's an `auto_map` in the `config`, which is
the same condition to add the `custom_code` tag to repos. This fixes:

* https://huggingface.co/allenai/OLMo-7B
* https://huggingface.co/togethercomputer/m2-bert-80M-2k-retrieval

---------

Co-authored-by: Julien Chaumond <[email protected]>

Author: osanseviero

packages/tasks/src/library-ui-elements.ts

@@ -1,6 +1,8 @@
 import type { ModelData } from "./model-data";
 import type { ModelLibraryKey } from "./model-libraries";
 
+const TAG_CUSTOM_CODE = "custom_code";
+
 /**
  * Elements configurable by a model library.
  */
@@ -422,7 +424,7 @@ const transformers = (model: ModelData) => {
 	if (!info) {
 		return [`# ⚠️ Type of model unknown`];
 	}
-	const remote_code_snippet = info.custom_class ? ", trust_remote_code=True" : "";
+	const remote_code_snippet = model.tags?.includes(TAG_CUSTOM_CODE) ? ", trust_remote_code=True" : "";
 
 	let autoSnippet: string;
 	if (info.processor) {