Update README.md

README.md

@@ -40,6 +40,7 @@ huggingface-cli download NewtNewt/MESA --local-dir ./weights
 ## Usage
 ```python
 from MESA.pipeline_terrain import TerrainDiffusionPipeline
+import torch
 
 pipe = TerrainDiffusionPipeline.from_pretrained("./weights", torch_dtype=torch.float16)
 pipe.to("cuda");