MCPcopy Create free account
hub / github.com/NVlabs/DiffusionNFT / collate_fn

Method collate_fn

scripts/train_nft_sd3.py:92–95  ·  view source on GitHub ↗
(examples)

Source from the content-addressed store, hash-verified

90
91 @staticmethod
92 def collate_fn(examples):
93 prompts = [example["prompt"] for example in examples]
94 metadatas = [example["metadata"] for example in examples]
95 return prompts, metadatas
96
97
98class GenevalPromptDataset(Dataset):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected