NCJ commited on
Commit
82cffcb
1 Parent(s): a5555ed

Update relighting_gen.py

Browse files
Files changed (1) hide show
  1. demo/relighting_gen.py +1 -1
demo/relighting_gen.py CHANGED
@@ -18,7 +18,7 @@ NeuralTextureControlNetModel = get_class_from_dynamic_module(
18
  "NeuralTextureControlNetModel"
19
  )
20
  controlnet = NeuralTextureControlNetModel.from_pretrained(
21
- "dilightnet/DiLightNet-submissions-300k",
22
  torch_dtype=dtype,
23
  )
24
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
 
18
  "NeuralTextureControlNetModel"
19
  )
20
  controlnet = NeuralTextureControlNetModel.from_pretrained(
21
+ "dilightnet/DiLightNet",
22
  torch_dtype=dtype,
23
  )
24
  pipe = StableDiffusionControlNetPipeline.from_pretrained(