Update README.md
Browse files
README.md
CHANGED
|
@@ -85,16 +85,16 @@ import torch
|
|
| 85 |
from diffusers import DiffusionPipeline
|
| 86 |
from optimum.quanto import freeze, quantize, qint8
|
| 87 |
|
| 88 |
-
model_id = "neuralvfx/LibreFlux-ControlNet"
|
| 89 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 90 |
dtype = torch.bfloat16 if device == "cuda" else torch.float32
|
| 91 |
|
| 92 |
pipe = DiffusionPipeline.from_pretrained(
|
| 93 |
model_id,
|
| 94 |
custom_pipeline=model_id,
|
| 95 |
-
trust_remote_code=True,
|
| 96 |
torch_dtype=dtype,
|
| 97 |
-
safety_checker=None
|
| 98 |
)
|
| 99 |
|
| 100 |
quantize(
|
|
|
|
| 85 |
from diffusers import DiffusionPipeline
|
| 86 |
from optimum.quanto import freeze, quantize, qint8
|
| 87 |
|
| 88 |
+
model_id = "neuralvfx/LibreFlux-ControlNet"
|
| 89 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 90 |
dtype = torch.bfloat16 if device == "cuda" else torch.float32
|
| 91 |
|
| 92 |
pipe = DiffusionPipeline.from_pretrained(
|
| 93 |
model_id,
|
| 94 |
custom_pipeline=model_id,
|
| 95 |
+
trust_remote_code=True,
|
| 96 |
torch_dtype=dtype,
|
| 97 |
+
safety_checker=None
|
| 98 |
)
|
| 99 |
|
| 100 |
quantize(
|