Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ You should use FRSCHIAN style to trigger the image generation.
|
|
| 32 |
```py
|
| 33 |
import torch
|
| 34 |
from diffusers import AutoPipelineForText2Image
|
| 35 |
-
pipe = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-
|
| 36 |
pipe.unload_lora_weights()
|
| 37 |
pipe.load_lora_weights("tarfandoon/farshchian_flux", weight_name="flux-lora.safetensors", adapter_name="fas-v2")
|
| 38 |
pipe.set_adapters(["fas-v2"], adapter_weights=[0.8])
|
|
|
|
| 32 |
```py
|
| 33 |
import torch
|
| 34 |
from diffusers import AutoPipelineForText2Image
|
| 35 |
+
pipe = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to("cuda")
|
| 36 |
pipe.unload_lora_weights()
|
| 37 |
pipe.load_lora_weights("tarfandoon/farshchian_flux", weight_name="flux-lora.safetensors", adapter_name="fas-v2")
|
| 38 |
pipe.set_adapters(["fas-v2"], adapter_weights=[0.8])
|