Tzktz
/

user-model-flux

Model card Files Files and versions

Tzktz commited on Dec 25, 2024

Commit

33d0a72

·

verified ·

1 Parent(s): d1aaf9c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ from diffusers import AutoPipelineForText2Image
 import torch
 pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
-pipeline.load_lora_weights('Tzktz/user-model-flux', weight_name='lora.safetensors')
 image = pipeline('your prompt').images[0]
 ```

 import torch
 pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
+pipeline.load_lora_weights('Tzktz/user-model-flux', weight_name='flux_train_codeplug.safetensors')
 image = pipeline('your prompt').images[0]
 ```