Spaces:
Running
Running
File size: 611 Bytes
2cd207c dbb7d2e 2cd207c dbb7d2e 2cd207c dbb7d2e 2cd207c 4d4f66e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from datasets import load_dataset, Dataset, load_from_disk
from diffusers import DPMSolverMultistepScheduler
DPMSolverMultistepScheduler.from_config()
def main():
promptbook = load_dataset('NYUSHPRP/ModelCofferPromptBook')
print(promptbook)
promptbook.save_to_disk('./promptbook')
roster = load_dataset('NYUSHPRP/ModelCofferRoster', split='train')
roster.save_to_disk('./roster')
def load():
roster = load_from_disk('./roster')
print(roster)
def test():
promptbook = load_from_disk('./promptbook')
print(promptbook[0]['image'])
if __name__ == '__main__':
main()
|