Spaces:

prithivMLmods
/

SD3.5-Turbo-Portrait

Running on Zero

prithivMLmods commited on 7 days ago

Commit

11e0d40

verified ·

1 Parent(s): 8e84f67

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,8 @@ torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
-pipe.load_lora_weights("strangerzonehf/SD3.5-Turbo-Portrait-LoRA", weight_name="SD3.5-Turbo-Portrait.safetensors")
 pipe.fuse_lora(lora_scale=1.0)
 MAX_SEED = np.iinfo(np.int32).max

 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
+pipe.load_lora_weights("strangerzonehf/NFT_Character_Avatar-SD3.5-TURBO-Large", weight_name="nft-avatar-3d-sd35.safetensors")
+trigger_word = "nft avatar 3d"  # Specify trigger word for LoRA
 pipe.fuse_lora(lora_scale=1.0)
 MAX_SEED = np.iinfo(np.int32).max