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
fb9e37ad
Commit
fb9e37ad
authored
Jun 15, 2022
by
Patrick von Platen
Browse files
correct logging
parent
273f9fee
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
15 additions
and
3 deletions
+15
-3
examples/train_ddpm.py
examples/train_ddpm.py
+1
-1
src/diffusers/pipelines/__init__.py
src/diffusers/pipelines/__init__.py
+5
-0
src/diffusers/pipelines/pipeline_glide.py
src/diffusers/pipelines/pipeline_glide.py
+9
-2
No files found.
examples/train_ddpm.py
View file @
fb9e37ad
...
...
@@ -9,10 +9,10 @@ from accelerate import Accelerator
from
datasets
import
load_dataset
from
diffusers
import
DDPM
,
DDPMScheduler
,
UNetModel
from
torchvision.transforms
import
(
CenterCrop
,
Compose
,
InterpolationMode
,
Lambda
,
CenterCrop
,
RandomHorizontalFlip
,
Resize
,
ToTensor
,
...
...
src/diffusers/pipelines/__init__.py
View file @
fb9e37ad
from
.pipeline_bddm
import
BDDM
from
.pipeline_ddim
import
DDIM
from
.pipeline_ddpm
import
DDPM
try
:
from
.pipeline_glide
import
GLIDE
except
ImportError
:
class
GLIDE
:
pass
from
.pipeline_latent_diffusion
import
LatentDiffusion
from
.pipeline_pndm
import
PNDM
src/diffusers/pipelines/pipeline_glide.py
View file @
fb9e37ad
...
...
@@ -15,7 +15,6 @@
""" PyTorch CLIP model."""
import
math
import
logging
from
dataclasses
import
dataclass
from
typing
import
Any
,
Optional
,
Tuple
,
Union
...
...
@@ -25,12 +24,19 @@ import torch.utils.checkpoint
from
torch
import
nn
import
tqdm
try
:
from
transformers
import
CLIPConfig
,
CLIPModel
,
CLIPTextConfig
,
CLIPVisionConfig
,
GPT2Tokenizer
from
transformers.activations
import
ACT2FN
from
transformers.modeling_outputs
import
BaseModelOutput
,
BaseModelOutputWithPooling
from
transformers.modeling_utils
import
PreTrainedModel
from
transformers.utils
import
ModelOutput
,
add_start_docstrings_to_model_forward
,
logging
,
replace_return_docstrings
from
transformers.utils
import
(
ModelOutput
,
add_start_docstrings_to_model_forward
,
logging
,
replace_return_docstrings
,
)
except
:
print
(
"Transformers is not installed"
)
pass
...
...
@@ -38,6 +44,7 @@ except:
from
..models
import
GLIDESuperResUNetModel
,
GLIDETextToImageUNetModel
from
..pipeline_utils
import
DiffusionPipeline
from
..schedulers
import
ClassifierFreeGuidanceScheduler
,
DDIMScheduler
from
..utils
import
logging
#####################
...
...
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