Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ snapshot_download(repo_id="BestWishYsh/ConsisID-preview", local_dir="BestWishYsh
|
|
41 |
model_path = "BestWishYsh/ConsisID-preview"
|
42 |
lora_path = None
|
43 |
lora_rank = 128
|
44 |
-
dtype = torch.
|
45 |
|
46 |
if os.path.exists(os.path.join(model_path, "transformer_ema")):
|
47 |
subfolder = "transformer_ema"
|
@@ -139,7 +139,7 @@ def delete_old_files():
|
|
139 |
if file_mtime < cutoff:
|
140 |
os.remove(file_path)
|
141 |
time.sleep(600)
|
142 |
-
@spaces.GPU(duration=
|
143 |
def plex(prompt,image_input,seed_value,scale_status,rife_status,progress=gr.Progress(track_tqdm=True)):
|
144 |
seed = seed_value
|
145 |
if seed == -1:
|
@@ -172,7 +172,7 @@ def plex(prompt,image_input,seed_value,scale_status,rife_status,progress=gr.Prog
|
|
172 |
prompt=prompt,
|
173 |
image=image,
|
174 |
num_videos_per_prompt=1,
|
175 |
-
num_inference_steps=
|
176 |
num_frames=49,
|
177 |
use_dynamic_cfg=False,
|
178 |
guidance_scale=7.0,
|
|
|
41 |
model_path = "BestWishYsh/ConsisID-preview"
|
42 |
lora_path = None
|
43 |
lora_rank = 128
|
44 |
+
dtype = torch.float16
|
45 |
|
46 |
if os.path.exists(os.path.join(model_path, "transformer_ema")):
|
47 |
subfolder = "transformer_ema"
|
|
|
139 |
if file_mtime < cutoff:
|
140 |
os.remove(file_path)
|
141 |
time.sleep(600)
|
142 |
+
@spaces.GPU(duration=120)
|
143 |
def plex(prompt,image_input,seed_value,scale_status,rife_status,progress=gr.Progress(track_tqdm=True)):
|
144 |
seed = seed_value
|
145 |
if seed == -1:
|
|
|
172 |
prompt=prompt,
|
173 |
image=image,
|
174 |
num_videos_per_prompt=1,
|
175 |
+
num_inference_steps=10,
|
176 |
num_frames=49,
|
177 |
use_dynamic_cfg=False,
|
178 |
guidance_scale=7.0,
|