From bc6c50964d1251f6d3e2a87e488db1ff763b6a2a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Fri, 3 Mar 2023 04:54:52 +0800 Subject: [PATCH] ckpt load --- pages/index.vue | 20 +++++++++++--------- server.py | 1 + 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/pages/index.vue b/pages/index.vue index d4695c2..81df0a2 100644 --- a/pages/index.vue +++ b/pages/index.vue @@ -9,10 +9,11 @@ div(class="mt-[60px] grid grid-cols-1 lg:grid-cols-4 xl:grid-cols-5 text-white b div.cursor-pointer.select-none(v-for="(item, index) in filters" :key="item.name" class="w-5/16 h-20 bg-gray-500 rounded-lg") div.flex.justify-center.items-center.h-full(v-if="!item.name" @click="ModelsShow(index, item)") img.w-6.h-6(:src="IconPlusCircle" alt="PlusCircle") - div.flex.items-end.h-full.border-purple-700.rounded-md( + // background-image: url(${item.image}); + div.flex.items-end.h-full.border-opacity-0.border-2.border-purple-700.rounded-md.transition-opacity.duration-150( v-else - :style="`background-image: url(${item.image}); background-size: cover; background-position: center;`" - :class="{'border-2': new_task.model === item.name}" + :style="`background-size: cover; background-position: center;`" + :class="{'border-opacity-100': new_task.ckpt === item.name}" @click="ModelsChange(item)" ) p(class="p-1") {{item.name}} @@ -228,12 +229,12 @@ const new_task = ref({ const filters = ref([ - { name:'None', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, - { name:'Colorpop1', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, - { name:'Colorpop2', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, - { name:'Colorpop3', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, - { name:'Colorpop4', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, - { name:'', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, + { name:'768-v-ema', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, + { name:'768-v-test', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, + //{ name:'baoxiang_2364', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, + //{ name:'mmk', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, + //{ name:'tx_2600', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, + { name:'', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' }, ]) const models = ref([ @@ -250,6 +251,7 @@ const models = ref([ const images = ref([]) const tasks = ref([]) const sizes = ref([ + { id:'image-dim-0', width:320, height:1024 }, { id:'image-dim-1', width:512, height:512 }, { id:'image-dim-2', width:768, height:768 }, { id:'image-dim-3', width:1024, height:1024 }, diff --git a/server.py b/server.py index c076ec1..bece1be 100644 --- a/server.py +++ b/server.py @@ -34,6 +34,7 @@ def load_model_from_config(config, ckpt, device=torch.device("cuda"), verbose=Fa pl_sd = torch.load(ckpt, map_location="cpu") if "global_step" in pl_sd: print(f"Global Step: {pl_sd['global_step']}") + print('torch: load ckpt done') sd = pl_sd["state_dict"] model = instantiate_from_config(config.model) m, u = model.load_state_dict(sd, strict=False)