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,
|