Spaces:
Running
on
L40S
Running
on
L40S
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":
|