Epoch 0

.gitignore

@@ -0,0 +1,2 @@
+step_*
+epoch_*

model_index.json

@@ -0,0 +1,12 @@
+{
+  "_class_name": "DDPMConditionPipeline",
+  "_diffusers_version": "0.16.1",
+  "scheduler": [
+    "diffusers",
+    "DDPMScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DModel"
+  ]
+}

params.json

@@ -0,0 +1,63 @@
+{
+    "dataset_name": [
+        "Mtot_Nbody_SIMBA"
+    ],
+    "dataset_config_name": null,
+    "dataset_conditional_name": null,
+    "model_config_name_or_path": null,
+    "base_channels": 64,
+    "cross_attention_dim": null,
+    "vae": null,
+    "vae_from_pretrained": null,
+    "vae_scaling_factor": null,
+    "train_data_dir": null,
+    "output_dir": "output/ddpm-128-epsilon-1000-linear-snr-run3",
+    "tag": [
+        "na1"
+    ],
+    "overwrite_output_dir": false,
+    "cache_dir": "data",
+    "resolution": 128,
+    "data_size": 13500,
+    "super_resolution": null,
+    "local_resize": false,
+    "conditional": false,
+    "center_crop": false,
+    "random_flip": false,
+    "train_batch_size": 64,
+    "eval_batch_size": 16,
+    "dataloader_num_workers": 0,
+    "num_epochs": 200,
+    "save_images_epochs": 10,
+    "save_model_epochs": 10,
+    "gradient_accumulation_steps": 1,
+    "learning_rate": 0.0001,
+    "lr_scheduler": "cosine",
+    "lr_warmup_steps": 500,
+    "adam_beta1": 0.95,
+    "adam_beta2": 0.999,
+    "adam_weight_decay": 1e-06,
+    "adam_epsilon": 1e-08,
+    "use_ema": true,
+    "ema_inv_gamma": 1.0,
+    "ema_power": 0.75,
+    "ema_max_decay": 0.9999,
+    "push_to_hub": true,
+    "hub_token": null,
+    "hub_model_id": null,
+    "hub_private_repo": false,
+    "logger": "wandb",
+    "logging_dir": "logs",
+    "local_rank": -1,
+    "mixed_precision": "no",
+    "prediction_type": "epsilon",
+    "loss": "mse",
+    "ddpm_num_steps": 1000,
+    "ddpm_num_inference_steps": 1000,
+    "ddpm_beta_schedule": "linear",
+    "ddpm_enforce_snr": true,
+    "checkpointing_steps": 10000,
+    "checkpoints_total_limit": null,
+    "resume_from_checkpoint": "latest",
+    "enable_xformers_memory_efficient_attention": false
+}

scheduler/scheduler_config.json

@@ -0,0 +1,1017 @@
+{
+  "_class_name": "DDPMScheduler",
+  "_diffusers_version": "0.16.1",
+  "beta_end": 0.02,
+  "beta_schedule": "linear",
+  "beta_start": 0.0001,
+  "clip_sample": true,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "sample_max_value": 1.0,
+  "thresholding": false,
+  "trained_betas": [
+    0.00010001659393310547,
+    0.00012063980102539062,
+    0.00014078617095947266,
+    0.00016075372695922852,
+    0.00018090009689331055,
+    0.0002009272575378418,
+    0.00022083520889282227,
+    0.0002409219741821289,
+    0.00026100873947143555,
+    0.00028121471405029297,
+    0.00030118227005004883,
+    0.0003210902214050293,
+    0.00034117698669433594,
+    0.0003612041473388672,
+    0.0003813505172729492,
+    0.0004013180732727051,
+    0.0004215240478515625,
+    0.0004413723945617676,
+    0.0004615187644958496,
+    0.00048148632049560547,
+    0.0005017518997192383,
+    0.000521540641784668,
+    0.00054168701171875,
+    0.0005616545677185059,
+    0.0005818009376525879,
+    0.0006018877029418945,
+    0.0006219148635864258,
+    0.0006420016288757324,
+    0.0006618499755859375,
+    0.0006820559501647949,
+    0.000702202320098877,
+    0.0007221698760986328,
+    0.0007421374320983887,
+    0.0007624030113220215,
+    0.0007821917533874512,
+    0.000802159309387207,
+    0.0008225440979003906,
+    0.0008423924446105957,
+    0.0008624792098999023,
+    0.0008826851844787598,
+    0.0009024739265441895,
+    0.0009226202964782715,
+    0.0009427666664123535,
+    0.0009626150131225586,
+    0.000982820987701416,
+    0.0010029077529907227,
+    0.0010228157043457031,
+    0.0010430216789245605,
+    0.0010629892349243164,
+    0.0010829567909240723,
+    0.0011031031608581543,
+    0.001123189926147461,
+    0.0011432766914367676,
+    0.0011632442474365234,
+    0.0011832714080810547,
+    0.0012034177780151367,
+    0.0012233853340148926,
+    0.00124359130859375,
+    0.0012635588645935059,
+    0.0012836456298828125,
+    0.0013036131858825684,
+    0.0013236403465270996,
+    0.0013437867164611816,
+    0.0013638138771057129,
+    0.0013837814331054688,
+    0.0014039874076843262,
+    0.001423954963684082,
+    0.0014440417289733887,
+    0.0014641284942626953,
+    0.0014840364456176758,
+    0.0015042424201965332,
+    0.0015243291854858398,
+    0.0015442967414855957,
+    0.0015644431114196777,
+    0.0015844106674194336,
+    0.0016044974327087402,
+    0.0016246438026428223,
+    0.0016446709632873535,
+    0.001664578914642334,
+    0.0016847848892211914,
+    0.0017047524452209473,
+    0.0017248988151550293,
+    0.001744985580444336,
+    0.0017650127410888672,
+    0.0017849206924438477,
+    0.0018053054809570312,
+    0.001825094223022461,
+    0.0018451809883117676,
+    0.001865386962890625,
+    0.0018854141235351562,
+    0.0019053816795349121,
+    0.0019254684448242188,
+    0.00194549560546875,
+    0.001965761184692383,
+    0.0019856691360473633,
+    0.0020058155059814453,
+    0.0020258426666259766,
+    0.0020458102226257324,
+    0.0020660758018493652,
+    0.0020859837532043457,
+    0.002106010913848877,
+    0.002126157283782959,
+    0.0021463632583618164,
+    0.002166450023651123,
+    0.0021863579750061035,
+    0.0022063851356506348,
+    0.002226531505584717,
+    0.0022466182708740234,
+    0.00226670503616333,
+    0.0022867321968078613,
+    0.002306818962097168,
+    0.0023270845413208008,
+    0.0023469924926757812,
+    0.002366781234741211,
+    0.00238722562789917,
+    0.002407252788543701,
+    0.0024272799491882324,
+    0.0024474263191223145,
+    0.0024673938751220703,
+    0.002487480640411377,
+    0.0025075674057006836,
+    0.0025276541709899902,
+    0.0025478601455688477,
+    0.002567768096923828,
+    0.0025878548622131348,
+    0.002608180046081543,
+    0.002627849578857422,
+    0.002648293972015381,
+    0.002668142318725586,
+    0.002688288688659668,
+    0.0027083754539489746,
+    0.002728581428527832,
+    0.0027486085891723633,
+    0.0027686357498168945,
+    0.002788841724395752,
+    0.0028087496757507324,
+    0.002828836441040039,
+    0.002849102020263672,
+    0.0028690695762634277,
+    0.002889275550842285,
+    0.002909362316131592,
+    0.0029291510581970215,
+    0.0029495954513549805,
+    0.0029695630073547363,
+    0.002989530563354492,
+    0.0030098557472229004,
+    0.0030298233032226562,
+    0.0030498504638671875,
+    0.0030701160430908203,
+    0.0030900239944458008,
+    0.003110229969024658,
+    0.003130316734313965,
+    0.0031504034996032715,
+    0.003170609474182129,
+    0.003190457820892334,
+    0.003210902214050293,
+    0.0032306909561157227,
+    0.00325089693069458,
+    0.003271043300628662,
+    0.003291189670562744,
+    0.0033111572265625,
+    0.0033313632011413574,
+    0.003351449966430664,
+    0.003371596336364746,
+    0.0033916831016540527,
+    0.003411591053009033,
+    0.003431975841522217,
+    0.0034519433975219727,
+    0.003472268581390381,
+    0.0034920573234558105,
+    0.0035125017166137695,
+    0.0035322904586791992,
+    0.003552556037902832,
+    0.0035726428031921387,
+    0.0035927891731262207,
+    0.003612995147705078,
+    0.0036330819129943848,
+    0.0036529898643493652,
+    0.003673374652862549,
+    0.0036933422088623047,
+    0.0037134289741516113,
+    0.003733813762664795,
+    0.003753662109375,
+    0.003773927688598633,
+    0.003794074058532715,
+    0.0038139820098876953,
+    0.003834366798400879,
+    0.0038542747497558594,
+    0.003874659538269043,
+    0.0038945674896240234,
+    0.003914773464202881,
+    0.003934800624847412,
+    0.003955185413360596,
+    0.003975093364715576,
+    0.003995418548583984,
+    0.004015564918518066,
+    0.004035472869873047,
+    0.00405573844909668,
+    0.004075944423675537,
+    0.004095971584320068,
+    0.004116058349609375,
+    0.004136383533477783,
+    0.00415647029876709,
+    0.0041765570640563965,
+    0.0041967034339904785,
+    0.004216969013214111,
+    0.004237055778503418,
+    0.004257023334503174,
+    0.004277408123016357,
+    0.0042975544929504395,
+    0.004317641258239746,
+    0.0043378472328186035,
+    0.004357695579528809,
+    0.004378318786621094,
+    0.004398167133331299,
+    0.004418373107910156,
+    0.00443875789642334,
+    0.004458725452423096,
+    0.0044789910316467285,
+    0.0044991374015808105,
+    0.004519164562225342,
+    0.004539430141448975,
+    0.004559576511383057,
+    0.004579663276672363,
+    0.004599928855895996,
+    0.004620075225830078,
+    0.004640340805053711,
+    0.004660546779632568,
+    0.0046805739402771,
+    0.004700779914855957,
+    0.0047209858894348145,
+    0.004741191864013672,
+    0.004761159420013428,
+    0.004781544208526611,
+    0.0048018693923950195,
+    0.004821956157684326,
+    0.004842102527618408,
+    0.00486224889755249,
+    0.004882335662841797,
+    0.004902839660644531,
+    0.0049228668212890625,
+    0.004942834377288818,
+    0.004963338375091553,
+    0.004983723163604736,
+    0.005003452301025391,
+    0.0050240159034729,
+    0.005043923854827881,
+    0.005064189434051514,
+    0.005084574222564697,
+    0.005104720592498779,
+    0.005124866962432861,
+    0.0051451921463012695,
+    0.00516510009765625,
+    0.00518566370010376,
+    0.005205929279327393,
+    0.005226016044616699,
+    0.0052460432052612305,
+    0.0052664875984191895,
+    0.005286812782287598,
+    0.005306601524353027,
+    0.005327165126800537,
+    0.00534743070602417,
+    0.005367398262023926,
+    0.005388021469116211,
+    0.005407989025115967,
+    0.005428194999694824,
+    0.005448400974273682,
+    0.005468785762786865,
+    0.005488932132720947,
+    0.005509316921234131,
+    0.005529344081878662,
+    0.005549728870391846,
+    0.005570113658905029,
+    0.005590200424194336,
+    0.005610406398773193,
+    0.005630791187286377,
+    0.005650877952575684,
+    0.005671143531799316,
+    0.005691587924957275,
+    0.005711853504180908,
+    0.00573199987411499,
+    0.005752265453338623,
+    0.005772590637207031,
+    0.005792796611785889,
+    0.005813181400299072,
+    0.005833566188812256,
+    0.005853533744812012,
+    0.005873978137969971,
+    0.005894184112548828,
+    0.005914628505706787,
+    0.00593489408493042,
+    0.005954921245574951,
+    0.00597536563873291,
+    0.005995810031890869,
+    0.006016075611114502,
+    0.006036281585693359,
+    0.006056487560272217,
+    0.006076633930206299,
+    0.0060974955558776855,
+    0.006117522716522217,
+    0.0061379075050354,
+    0.006157934665679932,
+    0.006178498268127441,
+    0.00619882345199585,
+    0.006219029426574707,
+    0.006239414215087891,
+    0.00625985860824585,
+    0.006279945373535156,
+    0.0063002705574035645,
+    0.0063210129737854,
+    0.006340980529785156,
+    0.006361186504364014,
+    0.006381869316101074,
+    0.006402134895324707,
+    0.006422460079193115,
+    0.006442844867706299,
+    0.006462991237640381,
+    0.006483674049377441,
+    0.0065038204193115234,
+    0.006524026393890381,
+    0.006544589996337891,
+    0.00656503438949585,
+    0.006585299968719482,
+    0.006605744361877441,
+    0.00662606954574585,
+    0.006646633148193359,
+    0.006666660308837891,
+    0.006687164306640625,
+    0.00670778751373291,
+    0.006727993488311768,
+    0.0067482590675354,
+    0.006768763065338135,
+    0.006789147853851318,
+    0.006809592247009277,
+    0.0068302154541015625,
+    0.006850063800811768,
+    0.00687098503112793,
+    0.006891310214996338,
+    0.006911575794219971,
+    0.006931900978088379,
+    0.00695270299911499,
+    0.006972968578338623,
+    0.0069931745529174805,
+    0.00701373815536499,
+    0.0070343017578125,
+    0.007054686546325684,
+    0.007075130939483643,
+    0.007095754146575928,
+    0.007115840911865234,
+    0.007136523723602295,
+    0.007157087326049805,
+    0.0071773529052734375,
+    0.007197737693786621,
+    0.007218718528747559,
+    0.007238566875457764,
+    0.007259488105773926,
+    0.007279753684997559,
+    0.007300376892089844,
+    0.007320702075958252,
+    0.007341265678405762,
+    0.007361710071563721,
+    0.0073822736740112305,
+    0.007402956485748291,
+    0.007423281669616699,
+    0.007443785667419434,
+    0.007464110851287842,
+    0.0074849724769592285,
+    0.007505297660827637,
+    0.007526040077209473,
+    0.007546484470367432,
+    0.007566928863525391,
+    0.007587611675262451,
+    0.007607996463775635,
+    0.007628679275512695,
+    0.00764918327331543,
+    0.007669925689697266,
+    0.007690310478210449,
+    0.007710874080657959,
+    0.0077315568923950195,
+    0.007752120494842529,
+    0.007772684097290039,
+    0.0077931880950927734,
+    0.007813870906829834,
+    0.007834434509277344,
+    0.007855057716369629,
+    0.007875680923461914,
+    0.007896244525909424,
+    0.007916867733001709,
+    0.00793766975402832,
+    0.007958173751831055,
+    0.00797879695892334,
+    0.007999300956726074,
+    0.008020102977752686,
+    0.00804072618484497,
+    0.008061468601226807,
+    0.008082091808319092,
+    0.008102595806121826,
+    0.008123397827148438,
+    0.008144140243530273,
+    0.008164823055267334,
+    0.00818544626235962,
+    0.00820612907409668,
+    0.008226871490478516,
+    0.008247554302215576,
+    0.008268117904663086,
+    0.008288979530334473,
+    0.008309900760650635,
+    0.00833040475845337,
+    0.008351147174835205,
+    0.008371949195861816,
+    0.008392512798309326,
+    0.008413374423980713,
+    0.008434176445007324,
+    0.00845479965209961,
+    0.008475899696350098,
+    0.008496403694152832,
+    0.008517265319824219,
+    0.008537888526916504,
+    0.008558988571166992,
+    0.008579611778259277,
+    0.008600413799285889,
+    0.008621394634246826,
+    0.008642077445983887,
+    0.008662819862365723,
+    0.008683860301971436,
+    0.00870448350906372,
+    0.008725583553314209,
+    0.008746147155761719,
+    0.008767247200012207,
+    0.00878816843032837,
+    0.00880885124206543,
+    0.008829891681671143,
+    0.008850514888763428,
+    0.008871793746948242,
+    0.008892416954040527,
+    0.00891333818435669,
+    0.008934319019317627,
+    0.008955299854278564,
+    0.008976161479949951,
+    0.008997082710266113,
+    0.009018182754516602,
+    0.009039103984832764,
+    0.009060144424438477,
+    0.009080946445465088,
+    0.0091019868850708,
+    0.009122967720031738,
+    0.009144127368927002,
+    0.009164750576019287,
+    0.009186208248138428,
+    0.009207069873809814,
+    0.009228110313415527,
+    0.009249210357666016,
+    0.009270250797271729,
+    0.009290993213653564,
+    0.009312272071838379,
+    0.009333610534667969,
+    0.009354472160339355,
+    0.009375512599945068,
+    0.009396672248840332,
+    0.009417951107025146,
+    0.009438753128051758,
+    0.009460031986236572,
+    0.009481251239776611,
+    0.009502410888671875,
+    0.009523510932922363,
+    0.009544730186462402,
+    0.009565889835357666,
+    0.00958704948425293,
+    0.009608328342437744,
+    0.009629547595977783,
+    0.009650886058807373,
+    0.009671807289123535,
+    0.0096932053565979,
+    0.009714305400848389,
+    0.009735822677612305,
+    0.009756922721862793,
+    0.009778261184692383,
+    0.009799718856811523,
+    0.009820699691772461,
+    0.009842216968536377,
+    0.009863674640655518,
+    0.009884953498840332,
+    0.009905993938446045,
+    0.009927690029144287,
+    0.009949028491973877,
+    0.009970247745513916,
+    0.009991645812988281,
+    0.010013401508331299,
+    0.010034561157226562,
+    0.010055959224700928,
+    0.01007753610610962,
+    0.010098636150360107,
+    0.0101204514503479,
+    0.010141968727111816,
+    0.010163605213165283,
+    0.010184884071350098,
+    0.010206222534179688,
+    0.01022791862487793,
+    0.010249435901641846,
+    0.010271191596984863,
+    0.010292589664459229,
+    0.010314226150512695,
+    0.010335862636566162,
+    0.010357558727264404,
+    0.010378837585449219,
+    0.010400652885437012,
+    0.010422229766845703,
+    0.010444104671478271,
+    0.010465741157531738,
+    0.010487616062164307,
+    0.010509252548217773,
+    0.010530710220336914,
+    0.010552644729614258,
+    0.01057422161102295,
+    0.010596215724945068,
+    0.010617733001708984,
+    0.010639607906341553,
+    0.0106620192527771,
+    0.010683119297027588,
+    0.01070547103881836,
+    0.010726988315582275,
+    0.01074904203414917,
+    0.010770916938781738,
+    0.010792851448059082,
+    0.010814845561981201,
+    0.01083672046661377,
+    0.010858476161956787,
+    0.010880708694458008,
+    0.010902762413024902,
+    0.01092463731765747,
+    0.01094675064086914,
+    0.01096886396408081,
+    0.01099085807800293,
+    0.011013031005859375,
+    0.011035144329071045,
+    0.011057138442993164,
+    0.011079251766204834,
+    0.01110166311264038,
+    0.0111236572265625,
+    0.011145830154418945,
+    0.011168241500854492,
+    0.011190235614776611,
+    0.011212825775146484,
+    0.011234879493713379,
+    0.01125723123550415,
+    0.011279702186584473,
+    0.011301815509796143,
+    0.01132434606552124,
+    0.011346757411956787,
+    0.011369049549102783,
+    0.01139158010482788,
+    0.011413931846618652,
+    0.011436522006988525,
+    0.011459052562713623,
+    0.01148158311843872,
+    0.011503934860229492,
+    0.01152646541595459,
+    0.011549293994903564,
+    0.011571884155273438,
+    0.011594295501708984,
+    0.011617183685302734,
+    0.011639595031738281,
+    0.011662662029266357,
+    0.011685192584991455,
+    0.01170802116394043,
+    0.011730611324310303,
+    0.011753618717193604,
+    0.011776447296142578,
+    0.011799216270446777,
+    0.011822044849395752,
+    0.011844992637634277,
+    0.011867940425872803,
+    0.011891007423400879,
+    0.01191401481628418,
+    0.011936843395233154,
+    0.011960029602050781,
+    0.011983096599578857,
+    0.012006044387817383,
+    0.012029290199279785,
+    0.012052476406097412,
+    0.01207578182220459,
+    0.01209867000579834,
+    0.012122094631195068,
+    0.012145400047302246,
+    0.012168407440185547,
+    0.012192070484161377,
+    0.012215495109558105,
+    0.012238740921020508,
+    0.012262165546417236,
+    0.012285709381103516,
+    0.01230919361114502,
+    0.012332499027252197,
+    0.012356102466583252,
+    0.012380063533782959,
+    0.012403428554534912,
+    0.012426912784576416,
+    0.012450754642486572,
+    0.012474536895751953,
+    0.012498140335083008,
+    0.01252204179763794,
+    0.012546122074127197,
+    0.012569665908813477,
+    0.012593686580657959,
+    0.01261758804321289,
+    0.012641549110412598,
+    0.012665510177612305,
+    0.012689709663391113,
+    0.01271367073059082,
+    0.012738168239593506,
+    0.01276165246963501,
+    0.012786448001861572,
+    0.01281040906906128,
+    0.012834668159484863,
+    0.012859225273132324,
+    0.012883305549621582,
+    0.012907981872558594,
+    0.012932538986206055,
+    0.012956559658050537,
+    0.012981653213500977,
+    0.01300579309463501,
+    0.013030529022216797,
+    0.013055145740509033,
+    0.013079941272735596,
+    0.013104498386383057,
+    0.01312953233718872,
+    0.01315450668334961,
+    0.01317906379699707,
+    0.01320427656173706,
+    0.013229310512542725,
+    0.013254106044769287,
+    0.013279438018798828,
+    0.013304531574249268,
+    0.013329505920410156,
+    0.013354957103729248,
+    0.013380110263824463,
+    0.013405561447143555,
+    0.01343071460723877,
+    0.013456404209136963,
+    0.01348203420639038,
+    0.013507187366485596,
+    0.013532876968383789,
+    0.013558685779571533,
+    0.013584375381469727,
+    0.01361018419265747,
+    0.013635873794555664,
+    0.01366192102432251,
+    0.013687849044799805,
+    0.0137137770652771,
+    0.01373988389968872,
+    0.013765931129455566,
+    0.013792097568511963,
+    0.013818562030792236,
+    0.013844609260559082,
+    0.013871192932128906,
+    0.013897538185119629,
+    0.013923883438110352,
+    0.013950645923614502,
+    0.013977289199829102,
+    0.014003872871398926,
+    0.014030754566192627,
+    0.014057576656341553,
+    0.014084339141845703,
+    0.014111459255218506,
+    0.014138340950012207,
+    0.014165401458740234,
+    0.014192700386047363,
+    0.014219939708709717,
+    0.014247238636016846,
+    0.014274835586547852,
+    0.01430201530456543,
+    0.014329671859741211,
+    0.014357209205627441,
+    0.01438528299331665,
+    0.01441270112991333,
+    0.014440834522247314,
+    0.014468610286712646,
+    0.01449662446975708,
+    0.014524996280670166,
+    0.014552712440490723,
+    0.01458132266998291,
+    0.01460963487625122,
+    0.014637887477874756,
+    0.01466667652130127,
+    0.014695286750793457,
+    0.014724016189575195,
+    0.014752686023712158,
+    0.014781594276428223,
+    0.014810740947723389,
+    0.014839589595794678,
+    0.01486903429031372,
+    0.014898180961608887,
+    0.01492762565612793,
+    0.014957010746002197,
+    0.014986574649810791,
+    0.015016257762908936,
+    0.01504594087600708,
+    0.015076220035552979,
+    0.015105843544006348,
+    0.01513594388961792,
+    0.01516646146774292,
+    0.01519632339477539,
+    0.015226960182189941,
+    0.015257656574249268,
+    0.015288233757019043,
+    0.015318632125854492,
+    0.015349864959716797,
+    0.015380680561065674,
+    0.015411853790283203,
+    0.015443027019500732,
+    0.015474259853363037,
+    0.01550602912902832,
+    0.015537440776824951,
+    0.015569150447845459,
+    0.015601098537445068,
+    0.015633046627044678,
+    0.015665113925933838,
+    0.015697240829467773,
+    0.015729904174804688,
+    0.0157623291015625,
+    0.01579493284225464,
+    0.015827655792236328,
+    0.01586073637008667,
+    0.01589357852935791,
+    0.01592695713043213,
+    0.015960276126861572,
+    0.015993952751159668,
+    0.01602727174758911,
+    0.016061484813690186,
+    0.016095280647277832,
+    0.016129374504089355,
+    0.01616346836090088,
+    0.01619821786880493,
+    0.016232669353485107,
+    0.01626765727996826,
+    0.01630234718322754,
+    0.01633751392364502,
+    0.016372978687286377,
+    0.016408205032348633,
+    0.01644420623779297,
+    0.016479551792144775,
+    0.016515731811523438,
+    0.016551852226257324,
+    0.016588211059570312,
+    0.016624808311462402,
+    0.016661643981933594,
+    0.01669830083847046,
+    0.016735732555389404,
+    0.016772985458374023,
+    0.01681041717529297,
+    0.016848325729370117,
+    0.016886234283447266,
+    0.016924262046813965,
+    0.016962885856628418,
+    0.01700127124786377,
+    0.017040133476257324,
+    0.01707911491394043,
+    0.017118394374847412,
+    0.01715785264968872,
+    0.01719754934310913,
+    0.01723766326904297,
+    0.017277657985687256,
+    0.017318129539489746,
+    0.01735895872116089,
+    0.017399609088897705,
+    0.0174410343170166,
+    0.017482280731201172,
+    0.017523884773254395,
+    0.017566144466400146,
+    0.017608225345611572,
+    0.0176507830619812,
+    0.01769334077835083,
+    0.01773667335510254,
+    0.017779767513275146,
+    0.017823755741119385,
+    0.01786738634109497,
+    0.017911434173583984,
+    0.017956137657165527,
+    0.018001019954681396,
+    0.018046140670776367,
+    0.018091559410095215,
+    0.018137097358703613,
+    0.018183648586273193,
+    0.018229663372039795,
+    0.018276751041412354,
+    0.01832348108291626,
+    0.01837092638015747,
+    0.01841902732849121,
+    0.018466830253601074,
+    0.018515586853027344,
+    0.018564283847808838,
+    0.018613755702972412,
+    0.01866316795349121,
+    0.018713295459747314,
+    0.018763601779937744,
+    0.0188143253326416,
+    0.018865764141082764,
+    0.018917322158813477,
+    0.01896911859512329,
+    0.01902174949645996,
+    0.01907438039779663,
+    0.01912778615951538,
+    0.019181668758392334,
+    0.019236087799072266,
+    0.019290447235107422,
+    0.01934570074081421,
+    0.01940131187438965,
+    0.01945728063583374,
+    0.019514083862304688,
+    0.01957082748413086,
+    0.019628524780273438,
+    0.019686758518218994,
+    0.01974540948867798,
+    0.019804418087005615,
+    0.019864380359649658,
+    0.019924461841583252,
+    0.019985198974609375,
+    0.020046710968017578,
+    0.02010875940322876,
+    0.02017110586166382,
+    0.020234525203704834,
+    0.02029818296432495,
+    0.020362257957458496,
+    0.020427942276000977,
+    0.02049332857131958,
+    0.02055966854095459,
+    0.020626544952392578,
+    0.020694613456726074,
+    0.02076268196105957,
+    0.0208318829536438,
+    0.020901739597320557,
+    0.020972132682800293,
+    0.021043777465820312,
+    0.021115779876708984,
+    0.02118849754333496,
+    0.02126210927963257,
+    0.021336793899536133,
+    0.021412014961242676,
+    0.021488070487976074,
+    0.021565020084381104,
+    0.021642863750457764,
+    0.021721720695495605,
+    0.021801412105560303,
+    0.02188187837600708,
+    0.021963417530059814,
+    0.02204573154449463,
+    0.022129356861114502,
+    0.022213637828826904,
+    0.022299230098724365,
+    0.02238595485687256,
+    0.022473394870758057,
+    0.022562265396118164,
+    0.022651970386505127,
+    0.02274245023727417,
+    0.0228348970413208,
+    0.02292865514755249,
+    0.023022472858428955,
+    0.02311837673187256,
+    0.023215293884277344,
+    0.023313581943511963,
+    0.023413240909576416,
+    0.023513972759246826,
+    0.023616135120391846,
+    0.023720622062683105,
+    0.023824870586395264,
+    0.023932218551635742,
+    0.02403968572616577,
+    0.024149775505065918,
+    0.024260997772216797,
+    0.02437382936477661,
+    0.024488210678100586,
+    0.02460455894470215,
+    0.024722158908843994,
+    0.024842143058776855,
+    0.024963736534118652,
+    0.02508687973022461,
+    0.02521216869354248,
+    0.02533942461013794,
+    0.025468647480010986,
+    0.025600075721740723,
+    0.025733351707458496,
+    0.025868654251098633,
+    0.026006579399108887,
+    0.026146769523620605,
+    0.02628910541534424,
+    0.02643340826034546,
+    0.02658092975616455,
+    0.026730597019195557,
+    0.026882529258728027,
+    0.02703791856765747,
+    0.02719557285308838,
+    0.027355730533599854,
+    0.0275193452835083,
+    0.02768552303314209,
+    0.027855277061462402,
+    0.028027057647705078,
+    0.028203189373016357,
+    0.028382301330566406,
+    0.028564810752868652,
+    0.02875065803527832,
+    0.028940141201019287,
+    0.029133975505828857,
+    0.029330849647521973,
+    0.029531896114349365,
+    0.029737532138824463,
+    0.029946446418762207,
+    0.030160486698150635,
+    0.03037923574447632,
+    0.030601799488067627,
+    0.03082960844039917,
+    0.031062424182891846,
+    0.03130018711090088,
+    0.03154337406158447,
+    0.03179210424423218,
+    0.03204619884490967,
+    0.03230631351470947,
+    0.03257250785827637,
+    0.03284496068954468,
+    0.033124327659606934,
+    0.03340965509414673,
+    0.033702731132507324,
+    0.03400254249572754,
+    0.034310221672058105,
+    0.03462541103363037,
+    0.03494906425476074,
+    0.03528094291687012,
+    0.03562188148498535,
+    0.035971879959106445,
+    0.0363314151763916,
+    0.036700546741485596,
+    0.03708028793334961,
+    0.03747093677520752,
+    0.0378727912902832,
+    0.03828620910644531,
+    0.03871220350265503,
+    0.03915059566497803,
+    0.03960263729095459,
+    0.04006928205490112,
+    0.040549516677856445,
+    0.041046202182769775,
+    0.041558265686035156,
+    0.0420876145362854,
+    0.04263538122177124,
+    0.04320061206817627,
+    0.04378700256347656,
+    0.044393301010131836,
+    0.04502171277999878,
+    0.045673668384552,
+    0.046349287033081055,
+    0.04705137014389038,
+    0.047780394554138184,
+    0.04853850603103638,
+    0.04932737350463867,
+    0.05014914274215698,
+    0.05100482702255249,
+    0.05189824104309082,
+    0.052830517292022705,
+    0.053804993629455566,
+    0.0548243522644043,
+    0.05589187145233154,
+    0.0570104718208313,
+    0.0581853985786438,
+    0.05941897630691528,
+    0.060716986656188965,
+    0.06208533048629761,
+    0.06352806091308594,
+    0.06505155563354492,
+    0.06666392087936401,
+    0.06837373971939087,
+    0.07018804550170898,
+    0.0721169114112854,
+    0.07417410612106323,
+    0.07637006044387817,
+    0.07871973514556885,
+    0.08124125003814697,
+    0.08395344018936157,
+    0.08687680959701538,
+    0.09004110097885132,
+    0.09347283840179443,
+    0.0972101092338562,
+    0.10129505395889282,
+    0.10577821731567383,
+    0.11072075366973877,
+    0.11619675159454346,
+    0.12229818105697632,
+    0.129136323928833,
+    0.13685625791549683,
+    0.14563679695129395,
+    0.15571177005767822,
+    0.16739320755004883,
+    0.1810930371284485,
+    0.19738423824310303,
+    0.21707504987716675,
+    0.24134713411331177,
+    0.271991491317749,
+    0.31187087297439575,
+    0.365816593170166,
+    0.44260871410369873,
+    0.5595889091491699,
+    0.7522659301757812,
+    1.0
+  ],
+  "variance_type": "fixed_small"
+}

unet/config.json

@@ -0,0 +1,46 @@
+{
+  "_class_name": "UNet2DModel",
+  "_diffusers_version": "0.16.1",
+  "act_fn": "silu",
+  "add_attention": true,
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    64,
+    64,
+    128,
+    128,
+    256,
+    256
+  ],
+  "center_input_sample": false,
+  "class_embed_type": null,
+  "down_block_types": [
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "DownBlock2D",
+    "AttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 1,
+  "layers_per_block": 2,
+  "mid_block_scale_factor": 1,
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "num_class_embeds": null,
+  "out_channels": 1,
+  "resnet_time_scale_shift": "default",
+  "sample_size": 128,
+  "time_embedding_type": "positional",
+  "up_block_types": [
+    "UpBlock2D",
+    "AttnUpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D",
+    "UpBlock2D"
+  ]
+}

unet/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a4bb03f4c5d17c0e1d7db9fd19f9a459d7cf0cd4053c237718e812051f00661
+size 113953789