Spaces:
Running
on
Zero
Running
on
Zero
minor changes
Browse files
app.py
CHANGED
@@ -18,10 +18,10 @@ good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfold
|
|
18 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, vae=taef1).to(device)
|
19 |
|
20 |
# PONIX mode load
|
21 |
-
pipe.load_lora_weights('cwhuh/ponix-generator-v0.1.0', weight_name='pytorch_lora_weights.safetensors')
|
22 |
-
embedding_path = hf_hub_download(repo_id='cwhuh/ponix-generator-v0.1.0', filename='./ponix-generator-v0.1.0_emb.safetensors', repo_type="model")
|
23 |
-
state_dict = load_file(embedding_path)
|
24 |
-
pipe.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>", "<s2>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
|
25 |
|
26 |
torch.cuda.empty_cache()
|
27 |
|
|
|
18 |
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype, vae=taef1).to(device)
|
19 |
|
20 |
# PONIX mode load
|
21 |
+
# pipe.load_lora_weights('cwhuh/ponix-generator-v0.1.0', weight_name='pytorch_lora_weights.safetensors')
|
22 |
+
# embedding_path = hf_hub_download(repo_id='cwhuh/ponix-generator-v0.1.0', filename='./ponix-generator-v0.1.0_emb.safetensors', repo_type="model")
|
23 |
+
# state_dict = load_file(embedding_path)
|
24 |
+
# pipe.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>", "<s2>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
|
25 |
|
26 |
torch.cuda.empty_cache()
|
27 |
|