Hugging Face's logo

Spaces:
chatham84
/
llama-2-13b-chat
Runtime error

App Files Community
1
chatham84 commited on
Commit
01d8b80
•
1 Parent(s): 1a322cc

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
6
 
7
  model_id = 'chatham84/llama-2-13b-chatham84-13b-64-8-1-v11'
8
-
9
 
10
  if torch.cuda.is_available():
11
  config = AutoConfig.from_pretrained(model_id)
 
5
  from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
6
 
7
  model_id = 'chatham84/llama-2-13b-chatham84-13b-64-8-1-v11'
8
+
9
 
10
  if torch.cuda.is_available():
11
  config = AutoConfig.from_pretrained(model_id)