Update README.md
Browse files
README.md
CHANGED
|
@@ -60,7 +60,7 @@ for index, item in enumerate(output):
|
|
| 60 |
from diffusers import AutoPipelineForText2Image
|
| 61 |
import torch
|
| 62 |
|
| 63 |
-
pipeline = AutoPipelineForText2Image.from_pretrained('
|
| 64 |
pipeline.load_lora_weights('codermert/yesim_fluxxx', weight_name='lora.safetensors')
|
| 65 |
image = pipeline('TOK').images[0]
|
| 66 |
```
|
|
|
|
| 60 |
from diffusers import AutoPipelineForText2Image
|
| 61 |
import torch
|
| 62 |
|
| 63 |
+
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
|
| 64 |
pipeline.load_lora_weights('codermert/yesim_fluxxx', weight_name='lora.safetensors')
|
| 65 |
image = pipeline('TOK').images[0]
|
| 66 |
```
|