scannet

app.py

@@ -69,21 +69,11 @@ class Preprocessor:
 
 torch.cuda.max_memory_allocated(device="cuda")
 
-# # Controlnet Normal
-# model_id = "lllyasviel/control_v11p_sd15_normalbae"
-# print("initializing controlnet")
-# controlnet = ControlNetModel.from_pretrained(
-#     model_id,
-#     torch_dtype=torch.float16,
-#     attn_implementation="flash_attention_2",
-# ).to("cuda")
-
 # Controlnet Normal
 model_id = "lllyasviel/control_v11p_sd15_normalbae"
 print("initializing controlnet")
-controlnet_file = hf_hub_download(model_id, filename="diffusion_pytorch_model.safetensors")
 controlnet = ControlNetModel.from_pretrained(
-    controlnet_file,
+    model_id,
     torch_dtype=torch.float16,
     attn_implementation="flash_attention_2",
 ).to("cuda")