JoPmt commited on
Commit
7cf6b76
1 Parent(s): 895b027

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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.bfloat16
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=70)
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=4,
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,