localoca5

failed to load model config file from colab

1

I failed when loading the model config file from colab. Code as below: #load model model = create_model(model_config).cpu() model.load_state_dict(load_state_dict(model_ckpt, location='cuda')) model = model.cuda() ddim_sampler = DDIMSampler(model) Got error: --------------------------------------------------------------------------- OSError...

如何输出更小的维度

1

您好！请问怎么让模型输出更小的维度呢(比如64维)？768维在实际中占用空间太大，而且计算也比较耗时