KingNish commited on
Commit
29476e7
1 Parent(s): 99c90b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ if torch.cuda.is_available():
105
  print("Using DALL-E 3 Consistency Decoder")
106
  pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
107
 
108
- if ENABLE_CPU_OFFLOAD:
109
  pipe.enable_model_cpu_offload()
110
  else:
111
  pipe.to(device)
 
105
  print("Using DALL-E 3 Consistency Decoder")
106
  pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
107
 
108
+ if ENABLE_CPU_OFFLOAD:
109
  pipe.enable_model_cpu_offload()
110
  else:
111
  pipe.to(device)