extra_networks_hypernet.py 620 B

123456789101112131415161718192021
  1. from modules import extra_networks
  2. from modules.hypernetworks import hypernetwork
  3. class ExtraNetworkHypernet(extra_networks.ExtraNetwork):
  4. def __init__(self):
  5. super().__init__('hypernet')
  6. def activate(self, p, params_list):
  7. names = []
  8. multipliers = []
  9. for params in params_list:
  10. assert len(params.items) > 0
  11. names.append(params.items[0])
  12. multipliers.append(float(params.items[1]) if len(params.items) > 1 else 1.0)
  13. hypernetwork.load_hypernetworks(names, multipliers)
  14. def deactivate(p, self):
  15. pass