Spaces:

EmanHassan26
/

ControlNetPromptGen

Running on Zero

EmanHassan26 commited on 22 days ago

Commit

064c240

verified ·

1 Parent(s): 9062672

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,8 +13,8 @@ from diffusers import ControlNetModel, StableDiffusionControlNetPipeline, UniPCM
 # Programmatically add the repository to PyTorch Hub's trusted list
 torch.hub.set_dir('/root/.cache/torch/hub') # Optional: ensures clean cache path
-torch.hub.trusted_list.append("rwightman/gen-efficientnet-pytorch")
-torch.hub.trusted_list.append("intel-isl/MiDaS")
 device = "cuda" if torch.cuda.is_available() else "cpu"

 # Programmatically add the repository to PyTorch Hub's trusted list
 torch.hub.set_dir('/root/.cache/torch/hub') # Optional: ensures clean cache path
+#torch.hub.trusted_list.append("rwightman/gen-efficientnet-pytorch")
+#torch.hub.trusted_list.append("intel-isl/MiDaS")
 device = "cuda" if torch.cuda.is_available() else "cpu"