undo generation.py changes

examples/llama_guard/generation.py

@@ -16,8 +16,8 @@ from fairscale.nn.model_parallel.initialize import (
     model_parallel_is_initialized,
 )
 
-from examples.llama_guard.model import ModelArgs, Transformer
-from examples.llama_guard.tokenizer import Tokenizer
+from llama_guard.model import ModelArgs, Transformer
+from llama_guard.tokenizer import Tokenizer
 
 Role = Literal["system", "user", "assistant"]