Skip to content

Hyphens-dashes converted to underscores for trigger words/file names #9

@ShaunCassidyPoster

Description

@ShaunCassidyPoster

Win 10; Intel Core i7 930 lol; 12 GB RAM; RTX 3060 12 GB.

Description: When creating a LoRA, projects named with hyphens, e.g. gumby-v1, will have the hyphen converted to underscore in the resulting safetensors file names: gumby_v1-step00000300.safetensors. At least one follow-on effect is that the Lora Manager custom node is unable to pull the trigger word into its TriggerWord Toggle node, and perhaps other nodes that explicitly associate the trigger word with the LoRA file name.

On generation, the Lora Manager TriggerWord Toggle node produces the error:

Traceback (most recent call last):
  File "F:\ComfyUI\ComfyUI\custom_nodes\comfyui-lora-manager\py\routes\lora_routes.py", line 148, in get_lora_trigger_words
    trigger_words = await self.service.get_lora_trigger_words(lora_name)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "F:\ComfyUI\ComfyUI\custom_nodes\comfyui-lora-manager\py\services\lora_service.py", line 260, in get_lora_trigger_words
    return civitai_data.get("trainedWords", [])
           ^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'get'

However, the LoRA apparently works. I say "apparently" because, although I use the hyphenated trigger word in the prompt, I am using Lora Manager and it may make using trigger words in the prompt unnecessary. I suppose I should know this, but I don't.

To Reproduce:

  • Create a dataset and project containing a hyphen
  • March through the training to create at least one checkpoint
  • Note that the trigger word part of the file name now has an underscore instead of a hyphen

Expected behavior:

  • I expected that part of the file name to be unchanged, since it is the basis of my dataset, prefixing every image and text file name.

Finally, despite my bug, thank you very much for making TrainFlow. LoRA training has always had a steep learning curve for a non-developer, and the time it would take to get up to speed was daunting. Seeing the likeness of my subject appear at step 600 was astounding. I hope TrainFlow catches on.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions