search

JoePenna / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable Diffusion (https://arxiv.org/abs/2112.10752). Tweaks focused on training faces, objects, and styles.
MIT License
3.2k stars 557 forks source link

Upload Images in Dreambooth Training Environment Setup fails on dreambooth_joepenna.ipynb #174

Closed devonuto closed 1 year ago

devonuto commented 1 year ago

Uploading training images...

AttributeError Traceback (most recent call last) /workspace/Dreambooth-Stable-Diffusion/JupyterNotebookHelpers/setup_training.py in submit_form_click(self, b) 194 return 195 else: --> 196 self.handle_training_images(uploaded_training_images) 197 198 # Regularization Images

/workspace/Dreambooth-Stable-Diffusion/JupyterNotebookHelpers/setup_training.py in handle_training_images(self, uploaded_images) 263 image_widgets = [] 264 for i, img in enumerate(uploaded_images): --> 265 images.append(img.name) 266 image_widgets.append(widgets.Image( 267 value=img.content,

AttributeError: 'str' object has no attribute 'name'

devonuto commented 1 year ago

Just using the training images upload button and selecting a bunch of images and hitting the Save button.

yushan777 commented 1 year ago

Are you using Runpod? I've seen this error if you use the runpod/pytorch-latest docker image template. Try using the runpod/pytorch-3.10-1.13.1-116 image instead.

More about this here: https://github.com/JoePenna/Dreambooth-Stable-Diffusion/discussions/170#discussion-5052317

devonuto commented 1 year ago

I was using runpod/pytorch:3.10-2.0.0-117

devonuto commented 1 year ago

@yushan777 thanks, that seems to have solved it.