Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
df2bc5ef
Unverified
Commit
df2bc5ef
authored
May 20, 2024
by
Sai-Suraj-27
Committed by
GitHub
May 20, 2024
Browse files
fix: Fixed few `docstrings` according to the Google Style Guide (#7717)
Fixed few docstrings according to the Google Style Guide.
parent
a7bf77fc
Changes
21
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
20 additions
and
20 deletions
+20
-20
examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
...diffusion_training/train_dreambooth_lora_sd15_advanced.py
+1
-1
examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
...diffusion_training/train_dreambooth_lora_sdxl_advanced.py
+1
-1
examples/community/stable_diffusion_reference.py
examples/community/stable_diffusion_reference.py
+1
-1
examples/custom_diffusion/train_custom_diffusion.py
examples/custom_diffusion/train_custom_diffusion.py
+1
-1
examples/dreambooth/train_dreambooth.py
examples/dreambooth/train_dreambooth.py
+1
-1
examples/dreambooth/train_dreambooth_flax.py
examples/dreambooth/train_dreambooth_flax.py
+1
-1
examples/dreambooth/train_dreambooth_lora.py
examples/dreambooth/train_dreambooth_lora.py
+1
-1
examples/dreambooth/train_dreambooth_lora_sdxl.py
examples/dreambooth/train_dreambooth_lora_sdxl.py
+1
-1
examples/research_projects/colossalai/train_dreambooth_colossalai.py
...search_projects/colossalai/train_dreambooth_colossalai.py
+1
-1
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
...h_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
+1
-1
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
...jects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
+1
-1
examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth.py
...eduled_huber_loss_training/dreambooth/train_dreambooth.py
+1
-1
examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth_lora.py
...d_huber_loss_training/dreambooth/train_dreambooth_lora.py
+1
-1
examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth_lora_sdxl.py
...er_loss_training/dreambooth/train_dreambooth_lora_sdxl.py
+1
-1
src/diffusers/pipelines/animatediff/pipeline_output.py
src/diffusers/pipelines/animatediff/pipeline_output.py
+1
-1
src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py
src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py
+1
-1
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py
...pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py
+1
-1
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py
...elines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py
+1
-1
src/diffusers/pipelines/shap_e/renderer.py
src/diffusers/pipelines/shap_e/renderer.py
+1
-1
src/diffusers/pipelines/text_to_video_synthesis/pipeline_output.py
...sers/pipelines/text_to_video_synthesis/pipeline_output.py
+1
-1
No files found.
examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py
View file @
df2bc5ef
...
@@ -981,7 +981,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -981,7 +981,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py
View file @
df2bc5ef
...
@@ -1136,7 +1136,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -1136,7 +1136,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/community/stable_diffusion_reference.py
View file @
df2bc5ef
...
@@ -78,7 +78,7 @@ def torch_dfs(model: torch.nn.Module):
...
@@ -78,7 +78,7 @@ def torch_dfs(model: torch.nn.Module):
class
StableDiffusionReferencePipeline
(
class
StableDiffusionReferencePipeline
(
DiffusionPipeline
,
TextualInversionLoaderMixin
,
LoraLoaderMixin
,
IPAdapterMixin
,
FromSingleFileMixin
DiffusionPipeline
,
TextualInversionLoaderMixin
,
LoraLoaderMixin
,
IPAdapterMixin
,
FromSingleFileMixin
):
):
r
"""
"
r
"""
Pipeline for Stable Diffusion Reference.
Pipeline for Stable Diffusion Reference.
This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods
This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods
...
...
examples/custom_diffusion/train_custom_diffusion.py
View file @
df2bc5ef
...
@@ -152,7 +152,7 @@ def collate_fn(examples, with_prior_preservation):
...
@@ -152,7 +152,7 @@ def collate_fn(examples, with_prior_preservation):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth.py
View file @
df2bc5ef
...
@@ -742,7 +742,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -742,7 +742,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth_flax.py
View file @
df2bc5ef
...
@@ -301,7 +301,7 @@ class DreamBoothDataset(Dataset):
...
@@ -301,7 +301,7 @@ class DreamBoothDataset(Dataset):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth_lora.py
View file @
df2bc5ef
...
@@ -680,7 +680,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -680,7 +680,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth_lora_sdxl.py
View file @
df2bc5ef
...
@@ -903,7 +903,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -903,7 +903,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/research_projects/colossalai/train_dreambooth_colossalai.py
View file @
df2bc5ef
...
@@ -327,7 +327,7 @@ class DreamBoothDataset(Dataset):
...
@@ -327,7 +327,7 @@ class DreamBoothDataset(Dataset):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
View file @
df2bc5ef
...
@@ -385,7 +385,7 @@ class DreamBoothDataset(Dataset):
...
@@ -385,7 +385,7 @@ class DreamBoothDataset(Dataset):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
View file @
df2bc5ef
...
@@ -384,7 +384,7 @@ class DreamBoothDataset(Dataset):
...
@@ -384,7 +384,7 @@ class DreamBoothDataset(Dataset):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth.py
View file @
df2bc5ef
...
@@ -762,7 +762,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -762,7 +762,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth_lora.py
View file @
df2bc5ef
...
@@ -700,7 +700,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -700,7 +700,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth_lora_sdxl.py
View file @
df2bc5ef
...
@@ -922,7 +922,7 @@ def collate_fn(examples, with_prior_preservation=False):
...
@@ -922,7 +922,7 @@ def collate_fn(examples, with_prior_preservation=False):
class
PromptDataset
(
Dataset
):
class
PromptDataset
(
Dataset
):
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
"A simple dataset to prepare the prompts to generate class images on multiple GPUs."
""
def
__init__
(
self
,
prompt
,
num_samples
):
def
__init__
(
self
,
prompt
,
num_samples
):
self
.
prompt
=
prompt
self
.
prompt
=
prompt
...
...
src/diffusers/pipelines/animatediff/pipeline_output.py
View file @
df2bc5ef
...
@@ -13,7 +13,7 @@ class AnimateDiffPipelineOutput(BaseOutput):
...
@@ -13,7 +13,7 @@ class AnimateDiffPipelineOutput(BaseOutput):
r
"""
r
"""
Output class for AnimateDiff pipelines.
Output class for AnimateDiff pipelines.
Args:
Args:
frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
denoised
denoised
...
...
src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py
View file @
df2bc5ef
...
@@ -76,7 +76,7 @@ class I2VGenXLPipelineOutput(BaseOutput):
...
@@ -76,7 +76,7 @@ class I2VGenXLPipelineOutput(BaseOutput):
r
"""
r
"""
Output class for image-to-video pipeline.
Output class for image-to-video pipeline.
Args:
Args:
frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
denoised
denoised
...
...
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py
View file @
df2bc5ef
...
@@ -1216,7 +1216,7 @@ class LEditsPPPipelineStableDiffusion(
...
@@ -1216,7 +1216,7 @@ class LEditsPPPipelineStableDiffusion(
Paper](https://arxiv.org/abs/2301.12247). If the scheduler is set to [`~schedulers.DDIMScheduler`] the
Paper](https://arxiv.org/abs/2301.12247). If the scheduler is set to [`~schedulers.DDIMScheduler`] the
inversion proposed by [edit-friendly DPDM](https://arxiv.org/abs/2304.06140) will be performed instead.
inversion proposed by [edit-friendly DPDM](https://arxiv.org/abs/2304.06140) will be performed instead.
Args:
Args:
image (`PipelineImageInput`):
image (`PipelineImageInput`):
Input for the image(s) that are to be edited. Multiple input images have to default to the same aspect
Input for the image(s) that are to be edited. Multiple input images have to default to the same aspect
ratio.
ratio.
...
...
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py
View file @
df2bc5ef
...
@@ -1449,7 +1449,7 @@ class LEditsPPPipelineStableDiffusionXL(
...
@@ -1449,7 +1449,7 @@ class LEditsPPPipelineStableDiffusionXL(
Paper](https://arxiv.org/abs/2301.12247). If the scheduler is set to [`~schedulers.DDIMScheduler`] the
Paper](https://arxiv.org/abs/2301.12247). If the scheduler is set to [`~schedulers.DDIMScheduler`] the
inversion proposed by [edit-friendly DPDM](https://arxiv.org/abs/2304.06140) will be performed instead.
inversion proposed by [edit-friendly DPDM](https://arxiv.org/abs/2304.06140) will be performed instead.
Args:
Args:
image (`PipelineImageInput`):
image (`PipelineImageInput`):
Input for the image(s) that are to be edited. Multiple input images have to default to the same aspect
Input for the image(s) that are to be edited. Multiple input images have to default to the same aspect
ratio.
ratio.
...
...
src/diffusers/pipelines/shap_e/renderer.py
View file @
df2bc5ef
...
@@ -844,7 +844,7 @@ class ShapERenderer(ModelMixin, ConfigMixin):
...
@@ -844,7 +844,7 @@ class ShapERenderer(ModelMixin, ConfigMixin):
transmittance(t[i + 1]) := transmittance(t[i]). 4) The last term is integration to infinity (e.g. [t[-1],
transmittance(t[i + 1]) := transmittance(t[i]). 4) The last term is integration to infinity (e.g. [t[-1],
math.inf]) that is evaluated by the void_model (i.e. we consider this space to be empty).
math.inf]) that is evaluated by the void_model (i.e. we consider this space to be empty).
a
rgs:
A
rgs:
rays: [batch_size x ... x 2 x 3] origin and direction. sampler: disjoint volume integrals. n_samples:
rays: [batch_size x ... x 2 x 3] origin and direction. sampler: disjoint volume integrals. n_samples:
number of ts to sample. prev_model_outputs: model outputs from the previous rendering step, including
number of ts to sample. prev_model_outputs: model outputs from the previous rendering step, including
...
...
src/diffusers/pipelines/text_to_video_synthesis/pipeline_output.py
View file @
df2bc5ef
...
@@ -15,7 +15,7 @@ class TextToVideoSDPipelineOutput(BaseOutput):
...
@@ -15,7 +15,7 @@ class TextToVideoSDPipelineOutput(BaseOutput):
"""
"""
Output class for text-to-video pipelines.
Output class for text-to-video pipelines.
Args:
Args:
frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):
List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
List of video outputs - It can be a nested list of length `batch_size,` with each sub-list containing
denoised
denoised
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment