multimodalart HF Staff commited on
Commit
2ce5689
·
verified ·
1 Parent(s): 44edc82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import os
10
 
11
  model_id = "decart-ai/Lucy-Edit-Dev"
12
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
13
- pipe = LucyEditPipeline.from_pretrained(model_id, vae, torch_dtype=torch.bfloat16)
14
  pipe.to("cuda")
15
 
16
  def calculate_resolution(input_width, input_height, max_dimension=832):
 
10
 
11
  model_id = "decart-ai/Lucy-Edit-Dev"
12
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
13
+ pipe = LucyEditPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
14
  pipe.to("cuda")
15
 
16
  def calculate_resolution(input_width, input_height, max_dimension=832):