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
Show 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
src/diffusers/pipelines/i2vgen_xl/pipeline_i2vgen_xl.py
View file @
df2bc5ef
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion.py
View file @
df2bc5ef
src/diffusers/pipelines/ledits_pp/pipeline_leditspp_stable_diffusion_xl.py
View file @
df2bc5ef
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
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