radames commited on
Commit
1af368e
1 Parent(s): dabf711

only vae unet

Browse files
Files changed (3) hide show
  1. app-controlnet.py +0 -2
  2. app-img2img.py +0 -2
  3. app-txt2img.py +0 -2
app-controlnet.py CHANGED
@@ -107,8 +107,6 @@ compel_proc = Compel(
107
  truncate_long_prompts=False,
108
  )
109
  if TORCH_COMPILE:
110
- pipe.text_encoder = torch.compile(pipe.text_encoder, mode="max-autotune", fullgraph=False)
111
- pipe.tokenizer = torch.compile(pipe.tokenizer, mode="max-autotune", fullgraph=False)
112
  pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
113
  pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
114
 
 
107
  truncate_long_prompts=False,
108
  )
109
  if TORCH_COMPILE:
 
 
110
  pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
111
  pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
112
 
app-img2img.py CHANGED
@@ -79,8 +79,6 @@ if psutil.virtual_memory().total < 64 * 1024**3:
79
  pipe.enable_attention_slicing()
80
 
81
  if TORCH_COMPILE:
82
- pipe.text_encoder = torch.compile(pipe.text_encoder, mode="max-autotune", fullgraph=False)
83
- pipe.tokenizer = torch.compile(pipe.tokenizer, mode="max-autotune", fullgraph=False)
84
  pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
85
  pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
86
 
 
79
  pipe.enable_attention_slicing()
80
 
81
  if TORCH_COMPILE:
 
 
82
  pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
83
  pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
84
 
app-txt2img.py CHANGED
@@ -79,8 +79,6 @@ if psutil.virtual_memory().total < 64 * 1024**3:
79
  pipe.enable_attention_slicing()
80
 
81
  if TORCH_COMPILE:
82
- pipe.text_encoder = torch.compile(pipe.text_encoder, mode="max-autotune", fullgraph=False)
83
- pipe.tokenizer = torch.compile(pipe.tokenizer, mode="max-autotune", fullgraph=False)
84
  pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
85
  pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
86
 
 
79
  pipe.enable_attention_slicing()
80
 
81
  if TORCH_COMPILE:
 
 
82
  pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
83
  pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
84