gameui/drawing
2
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

ckpt load

Browse Source
This commit is contained in:
桜華
2023-03-03 04:54:52 +08:00
parent 75274a9015
commit bc6c50964d
2 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pages/index.vue
View File

@@ -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.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)") 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") 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 v-else
:style="`background-image: url(${item.image}); background-size: cover; background-position: center;`" :style="`background-size: cover; background-position: center;`"
:class="{'border-2': new_task.model === item.name}" :class="{'border-opacity-100': new_task.ckpt === item.name}"
@click="ModelsChange(item)" @click="ModelsChange(item)"
) )
p(class="p-1") {{item.name}} p(class="p-1") {{item.name}}
@@ -228,12 +229,12 @@ const new_task = ref({
const filters = ref([ const filters = ref([
{ name:'None', 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:'Colorpop1', 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:'Colorpop2', 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:'Colorpop3', 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:'Colorpop4', 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' }, { name:'', image:'https://storage.googleapis.com/pai-marketing/filters/elizaport_style.png' },
]) ])
const models = ref([ const models = ref([
@@ -250,6 +251,7 @@ const models = ref([
const images = ref([]) const images = ref([])
const tasks = ref([]) const tasks = ref([])
const sizes = ref([ const sizes = ref([
{ id:'image-dim-0', width:320, height:1024 },
{ id:'image-dim-1', width:512, height:512 }, { id:'image-dim-1', width:512, height:512 },
{ id:'image-dim-2', width:768, height:768 }, { id:'image-dim-2', width:768, height:768 },
{ id:'image-dim-3', width:1024, height:1024 }, { id:'image-dim-3', width:1024, height:1024 },

1
server.py
View File

@@ -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") pl_sd = torch.load(ckpt, map_location="cpu")
if "global_step" in pl_sd: if "global_step" in pl_sd:
print(f"Global Step: {pl_sd['global_step']}") print(f"Global Step: {pl_sd['global_step']}")
print('torch: load ckpt done')
sd = pl_sd["state_dict"] sd = pl_sd["state_dict"]
model = instantiate_from_config(config.model) model = instantiate_from_config(config.model)
m, u = model.load_state_dict(sd, strict=False) m, u = model.load_state_dict(sd, strict=False)