neuralvfx commited on
Commit
0c97cea
·
verified ·
1 Parent(s): 0ffc2f0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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" # <-- change if you used a different repo
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, # required for your custom pipeline + modules
96
  torch_dtype=dtype,
97
- safety_checker=None # optional; disable if not using a checker
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(