Update README.md
Browse files
README.md
CHANGED
@@ -73,7 +73,7 @@ Weights for this model are available in Safetensors format.
|
|
73 |
from diffusers import AutoPipelineForText2Image
|
74 |
import torch
|
75 |
|
76 |
-
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.
|
77 |
pipeline.load_lora_weights('multimodalart/flux-tarot-v1', weight_name='flux_tarot_v1_lora.safetensors')
|
78 |
image = pipeline('in the style of TOK a trtcrd tarot style').images[0]
|
79 |
```
|
|
|
73 |
from diffusers import AutoPipelineForText2Image
|
74 |
import torch
|
75 |
|
76 |
+
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
|
77 |
pipeline.load_lora_weights('multimodalart/flux-tarot-v1', weight_name='flux_tarot_v1_lora.safetensors')
|
78 |
image = pipeline('in the style of TOK a trtcrd tarot style').images[0]
|
79 |
```
|