NCJ commited on
Commit
8ef729c
1 Parent(s): f286cb8

Update demo/relighting_gen.py

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