o e@sddlZddlmZddlZddlmZddlmZm Z dZ e dj dej dZeje ej_ejZejZej d eed j dej dZeje ej_ej d eed j dej dZeje ej_ej d dd dj dej dZe ejje_eddZdgdgdgdgdgdgdgdgdgdgdgdgdgdggZd Zejed!Ze-e d"e!ej"d#d$j#d%d&d'-e$ej%d(d%d)d(d%d*Z&Wdn1swYej'd+dd,Z(Wdn1swYe"ej$d-d.Z)ejd%d/Z*e d0Z+Wdn 1swYej$d-d.Z,ejd%d/Z-e d1Z.Wdn 1szopenMUSE/muse-256cuda)dtypezopenMUSE/muse-512) text_encodervaezopenMUSE/muse-512-finetunedzrunwayml/stable-diffusion-v1-5fp16)safety_checkervariant) torch_dtypecCstdt|ddddddd}td t|ddd dddd}td t|ddd dddd}td t|d djd}||||g}|S)NzGenerating 256: T)r) timestepsguidance_scaletransformer_seq_lenuse_fp16 temperaturerzGenerating 512:izGenerating 512 fine-tuned:zGenerating SD v15 fine-tuned:)num_inference_steps)printmuse_256muse_512 muse_512_finesdv1_5images)promptmuse_256_imagemuse_512_imagemuse_512_fine_image sdv1_5_imagerr$#/home/patrick/MUSE-vs-SD.1.5/app.pyinfers,    r&zKA small cabin on top of a snowy mountain in the style of Disney, artstationz a monkey doing yoga on the beachz'half human half cat, a human cat hybridza hedgehog using a calculatorzkanye west | diffuse lighting | fantasy | intricate elegant highly detailed lifelike photorealistic digital painting | artstationz astronaut pigz!two people shouting at each otherz(A linked in profile picture of Elon Muskz#A man looking out of a rainy windowzkclose up, iron man, eating breakfast in a cabin, symmetrical balance, hyper-realistic --ar 16:9 --style rawz5A high tech solarpunk utopia in the Amazon rainforestz5A pikachu fine dining with a view to the Eiffel Towerz0A mecha robot in a favela in expressionist stylez*an insect robot preparing a delicious mealzT h1 { text-align: center; } #component-0 { max-width: 730px; margin: auto; } )csszSDv15 vs. MUSE.zprompt-container)elem_idFT)mobile_collapse equal_heightzEnter your prompt)label show_label max_lines placeholder containerzGenerate image)scaler) min_width) interactivezSD V1-5z MUSE-Base-256z MUSE-Base-512zMUSE-Finetuned-512)examplesfninputsoutputscache_examples)r6r7)share);gradiogrPILrtorchmuser diffusersrr PAD_TOKENfrom_pretrainedtofloat16r transformer*enable_xformers_memory_efficient_attention tokenizer pad_tokenrr rrr from_config schedulerconfigr&r4r'BlocksblockMarkdownGroupRowstyleColumnTextboxtextButtonbtnc1image_1 image_1_labelc2image_2 image_2_labelimage_3 image_3_labelc3image_4 image_4_labelExamplesexdatasetheaderssubmitclicklaunchr$r$r$r%s     .              "