Flux-Character-Consistancy

Running on Zero

zs38 commited on Jan 20

Commit

f763f30

1 Parent(s): 591cb42

update

Files changed (2) hide show

app.py CHANGED Viewed

@@ -25,6 +25,7 @@ conditioner_base_model = 'eva02_large_patch14_448.mim_in22k_ft_in1k'
 conditioner_layer_num = 12
 device = "cuda" if torch.cuda.is_available() else "cpu"
 output_dim = 4096
 logger.info("init model")
 model = FluxTransformer2DModelWithIP.from_config(model_config, torch_dtype=torch.bfloat16) # type: ignore

 conditioner_layer_num = 12
 device = "cuda" if torch.cuda.is_available() else "cpu"
 output_dim = 4096
+logger.info(f"pretrained_model_name: {pretrained_model_name}, adapter_repo_id: {adapter_repo_id}, adapter_path: {adapter_path}, conditioner_layer: {conditioner_layer_num}, output_dim {output_dim}, device: {device}")
 logger.info("init model")
 model = FluxTransformer2DModelWithIP.from_config(model_config, torch_dtype=torch.bfloat16) # type: ignore

requirements.txt CHANGED Viewed

@@ -6,4 +6,5 @@ transformers
 torchvision
 timm
 loguru
-einops

 torchvision
 timm
 loguru
+einops
+sentencepiece