Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ if resume:
|
|
67 |
|
68 |
deviceType = 'cuda' if torch.cuda.is_available() else 'cpu'
|
69 |
m = model.to(deviceType)
|
70 |
-
tokenizer_gpt = Tokenizer(checkpoint_dir=Path("checkpoints
|
71 |
|
72 |
|
73 |
def inference(input_context, count):
|
|
|
67 |
|
68 |
deviceType = 'cuda' if torch.cuda.is_available() else 'cpu'
|
69 |
m = model.to(deviceType)
|
70 |
+
tokenizer_gpt = Tokenizer(checkpoint_dir=Path("checkpoints/meta-llama/Llama-2-7b-chat-hf"))
|
71 |
|
72 |
|
73 |
def inference(input_context, count):
|