@@ -132,30 +132,36 @@ depth = model.infer_image(raw_img) # HxW raw depth map in numpy
132
132
const diffusers_default = ( model : ModelData ) => [
133
133
`from diffusers import DiffusionPipeline
134
134
135
- pipeline = DiffusionPipeline.from_pretrained("${ model . id } ")` ,
135
+ pipe = DiffusionPipeline.from_pretrained("${ model . id } ")
136
+
137
+ prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
138
+ image = pipe(prompt).images[0]` ,
136
139
] ;
137
140
138
141
const diffusers_controlnet = ( model : ModelData ) => [
139
142
`from diffusers import ControlNetModel, StableDiffusionControlNetPipeline
140
143
141
144
controlnet = ControlNetModel.from_pretrained("${ model . id } ")
142
- pipeline = StableDiffusionControlNetPipeline.from_pretrained(
145
+ pipe = StableDiffusionControlNetPipeline.from_pretrained(
143
146
"${ get_base_diffusers_model ( model ) } ", controlnet=controlnet
144
147
)` ,
145
148
] ;
146
149
147
150
const diffusers_lora = ( model : ModelData ) => [
148
151
`from diffusers import DiffusionPipeline
149
152
150
- pipeline = DiffusionPipeline.from_pretrained("${ get_base_diffusers_model ( model ) } ")
151
- pipeline.load_lora_weights("${ model . id } ")` ,
153
+ pipe = DiffusionPipeline.from_pretrained("${ get_base_diffusers_model ( model ) } ")
154
+ pipe.load_lora_weights("${ model . id } ")
155
+
156
+ prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
157
+ image = pipe(prompt).images[0]` ,
152
158
] ;
153
159
154
160
const diffusers_textual_inversion = ( model : ModelData ) => [
155
161
`from diffusers import DiffusionPipeline
156
162
157
- pipeline = DiffusionPipeline.from_pretrained("${ get_base_diffusers_model ( model ) } ")
158
- pipeline .load_textual_inversion("${ model . id } ")`,
163
+ pipe = DiffusionPipeline.from_pretrained("${ get_base_diffusers_model ( model ) } ")
164
+ pipe .load_textual_inversion("${ model . id } ")`,
159
165
] ;
160
166
161
167
export const diffusers = ( model : ModelData ) : string [ ] => {
0 commit comments