fffiloni commited on
Commit
8b95bc3
1 Parent(s): ef5add3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ transformer = CogVideoXTransformer3DModel.from_pretrained(model_id, subfolder="t
31
  text_encoder = T5EncoderModel.from_pretrained(model_id, subfolder="text_encoder", torch_dtype=torch.float16)
32
  vae = AutoencoderKLCogVideoX.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float16)
33
  tokenizer = T5Tokenizer.from_pretrained(model_id, subfolder="tokenizer")
34
- pipe = CogVideoXImageToVideoPipeline.from_pretrained(model_id, tokenizer=tokenizer, text_encoder=text_encoder, transformer=transformer, vae=vae, torch_dtype=torch.float16)
35
 
36
  def find_and_move_object_to_cpu():
37
  for obj in gc.get_objects():
@@ -73,7 +73,7 @@ def clear_gpu():
73
 
74
 
75
  def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True)):
76
-
77
  lora_path = "checkpoints/"
78
  adapter_name = None
79
  if orbit_type == "Left":
 
31
  text_encoder = T5EncoderModel.from_pretrained(model_id, subfolder="text_encoder", torch_dtype=torch.float16)
32
  vae = AutoencoderKLCogVideoX.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float16)
33
  tokenizer = T5Tokenizer.from_pretrained(model_id, subfolder="tokenizer")
34
+
35
 
36
  def find_and_move_object_to_cpu():
37
  for obj in gc.get_objects():
 
73
 
74
 
75
  def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True)):
76
+ pipe = CogVideoXImageToVideoPipeline.from_pretrained(model_id, tokenizer=tokenizer, text_encoder=text_encoder, transformer=transformer, vae=vae, torch_dtype=torch.float16)
77
  lora_path = "checkpoints/"
78
  adapter_name = None
79
  if orbit_type == "Left":