dylanebert HF staff commited on
Commit
a283418
1 Parent(s): a3fbb44
Files changed (1) hide show
  1. pipeline.py +14 -27
pipeline.py CHANGED
@@ -1,39 +1,26 @@
1
- import torch
2
- import torch.nn.functional as F
3
  import inspect
4
- import numpy as np
5
- from typing import Callable, List, Optional, Union, Any
6
- from transformers import (
7
- CLIPTextModel,
8
- CLIPTokenizer,
9
- CLIPVisionModel,
10
- CLIPImageProcessor,
11
- )
12
- from diffusers import AutoencoderKL, DiffusionPipeline
13
- from diffusers.utils import (
14
- deprecate,
15
- is_accelerate_available,
16
- is_accelerate_version,
17
- logging,
18
- )
19
- from diffusers.configuration_utils import FrozenDict
20
- from diffusers.schedulers import DDIMScheduler
21
- from diffusers.utils.torch_utils import randn_tensor
22
-
23
  import math
24
  from inspect import isfunction
 
25
 
 
 
26
  import torch.nn as nn
27
- from einops import rearrange, repeat
28
-
29
- from diffusers.configuration_utils import ConfigMixin
30
- from diffusers.models.modeling_utils import ModelMixin
31
-
32
  # require xformers!
33
  import xformers
34
  import xformers.ops
35
-
 
 
 
 
 
 
 
36
  from kiui.cam import orbit_camera
 
 
37
 
38
 
39
  def get_camera(
 
 
 
1
  import inspect
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  import math
3
  from inspect import isfunction
4
+ from typing import Any, Callable, List, Optional, Union
5
 
6
+ import numpy as np
7
+ import torch
8
  import torch.nn as nn
9
+ import torch.nn.functional as F
 
 
 
 
10
  # require xformers!
11
  import xformers
12
  import xformers.ops
13
+ from diffusers import AutoencoderKL, DiffusionPipeline
14
+ from diffusers.configuration_utils import ConfigMixin, FrozenDict
15
+ from diffusers.models.modeling_utils import ModelMixin
16
+ from diffusers.schedulers import DDIMScheduler
17
+ from diffusers.utils import (deprecate, is_accelerate_available,
18
+ is_accelerate_version, logging)
19
+ from diffusers.utils.torch_utils import randn_tensor
20
+ from einops import rearrange, repeat
21
  from kiui.cam import orbit_camera
22
+ from transformers import (CLIPImageProcessor, CLIPTextModel, CLIPTokenizer,
23
+ CLIPVisionModel)
24
 
25
 
26
  def get_camera(