AWPortraitCN-2

Paused

DamarJati commited on Apr 3

Commit

7c45b36

verified ·

1 Parent(s): f324a39

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload()
 # pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.load_lora_weights("Shakker-Labs/AWPortraitCN2", weight_name="AWPortraitCN_2.safetensors", low_cpu_mem_usage=True)
 pipe.fuse_lora()
 MAX_SEED = 2**32-1

 pipe.enable_model_cpu_offload()
 # pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
+pipe.load_lora_weights("Shakker-Labs/AWPortraitCN2", weight_name="AWPortraitCN_2.safetensors")
 pipe.fuse_lora()
 MAX_SEED = 2**32-1