Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
| 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":
|