eggarsway commited on
Commit
3a84324
1 Parent(s): 60e276e
DirectedDiffusion/AttnEditorUtils.py CHANGED
@@ -62,14 +62,12 @@ def load_all_models(model_path_diffusion):
62
  unet = UNet2DConditionModel.from_pretrained(
63
  model_path_diffusion,
64
  subfolder="unet",
65
- use_auth_token=auth_token,
66
  revision="fp16",
67
  torch_dtype=torch.float16,
68
  )
69
  vae = AutoencoderKL.from_pretrained(
70
  model_path_diffusion,
71
  subfolder="vae",
72
- use_auth_token=auth_token,
73
  revision="fp16",
74
  torch_dtype=torch.float16,
75
  )
62
  unet = UNet2DConditionModel.from_pretrained(
63
  model_path_diffusion,
64
  subfolder="unet",
 
65
  revision="fp16",
66
  torch_dtype=torch.float16,
67
  )
68
  vae = AutoencoderKL.from_pretrained(
69
  model_path_diffusion,
70
  subfolder="vae",
 
71
  revision="fp16",
72
  torch_dtype=torch.float16,
73
  )