ConsisID

Running on Zero

JoPmt commited on 16 days ago

Commit

251c2ed

•

1 Parent(s): 7cf6b76

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ snapshot_download(repo_id="BestWishYsh/ConsisID-preview", local_dir="BestWishYsh
 model_path = "BestWishYsh/ConsisID-preview"
 lora_path = None
 lora_rank = 128
-dtype = torch.float16
 if os.path.exists(os.path.join(model_path, "transformer_ema")):
     subfolder = "transformer_ema"

 model_path = "BestWishYsh/ConsisID-preview"
 lora_path = None
 lora_rank = 128
+dtype = torch.bfloat16
 if os.path.exists(os.path.join(model_path, "transformer_ema")):
     subfolder = "transformer_ema"