Spaces:
Build error
Build error
from ldm.util import instantiate_from_config | |
def get_state_dict(d): | |
return d.get('state_dict', d) | |
def create_model(config, **kwargs): | |
model = instantiate_from_config(config.model).cpu() | |
return model | |