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):
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
):
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):
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
):
self
.
prompt
=
prompt
...
...
examples/community/stable_diffusion_reference.py
View file @
df2bc5ef
...
...
@@ -78,7 +78,7 @@ def torch_dfs(model: torch.nn.Module):
class
StableDiffusionReferencePipeline
(
DiffusionPipeline
,
TextualInversionLoaderMixin
,
LoraLoaderMixin
,
IPAdapterMixin
,
FromSingleFileMixin
):
r
"""
"
r
"""
Pipeline for Stable Diffusion Reference.
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):
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
):
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth.py
View file @
df2bc5ef
...
...
@@ -742,7 +742,7 @@ def collate_fn(examples, with_prior_preservation=False):
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
):
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth_flax.py
View file @
df2bc5ef
...
...
@@ -301,7 +301,7 @@ class DreamBoothDataset(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
):
self
.
prompt
=
prompt
...
...
examples/dreambooth/train_dreambooth_lora.py
View file @
df2bc5ef
...
...
@@ -680,7 +680,7 @@ def collate_fn(examples, with_prior_preservation=False):
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
):
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):
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
):
self
.
prompt
=
prompt
...
...
examples/research_projects/colossalai/train_dreambooth_colossalai.py
View file @
df2bc5ef
...
...
@@ -327,7 +327,7 @@ class DreamBoothDataset(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
):
self
.
prompt
=
prompt
...
...
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
View file @
df2bc5ef
...
...
@@ -385,7 +385,7 @@ class DreamBoothDataset(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
):
self
.
prompt
=
prompt
...
...
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py
View file @
df2bc5ef
...
...
@@ -384,7 +384,7 @@ class DreamBoothDataset(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
):
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):
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
):
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):
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
):
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):
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
):
self
.
prompt
=
prompt
...
...
src/diffusers/pipelines/animatediff/pipeline_output.py
View file @
df2bc5ef
...
...
@@ -13,7 +13,7 @@ class AnimateDiffPipelineOutput(BaseOutput):
r
"""
Output class for AnimateDiff pipelines.
Args:
Args:
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
denoised
...
...
src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py
View file @
df2bc5ef
...
...
@@ -76,7 +76,7 @@ class I2VGenXLPipelineOutput(BaseOutput):
r
"""
Output class for image-to-video pipeline.
Args:
Args:
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
denoised
...
...
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py
View file @
df2bc5ef
...
...
@@ -1216,7 +1216,7 @@ class LEditsPPPipelineStableDiffusion(
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.
Args:
Args:
image (`PipelineImageInput`):
Input for the image(s) that are to be edited. Multiple input images have to default to the same aspect
ratio.
...
...
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py
View file @
df2bc5ef
...
...
@@ -1449,7 +1449,7 @@ class LEditsPPPipelineStableDiffusionXL(
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.
Args:
Args:
image (`PipelineImageInput`):
Input for the image(s) that are to be edited. Multiple input images have to default to the same aspect
ratio.
...
...
src/diffusers/pipelines/shap_e/renderer.py
View file @
df2bc5ef
...
...
@@ -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],
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:
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):
"""
Output class for text-to-video pipelines.
Args:
Args:
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
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