KuangDW commited on
Commit
ba00803
·
1 Parent(s): 5e39340

change to Llama-3.3-70B-Instruct

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
61
  print(f"Using device: {device}")
62
  # Load models once
63
  print("Loading models...")
64
- model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
65
  tokenizer = AutoTokenizer.from_pretrained(model_id)
66
  model = AutoModelForCausalLM.from_pretrained(
67
  model_id,
 
61
  print(f"Using device: {device}")
62
  # Load models once
63
  print("Loading models...")
64
+ model_id = "meta-llama/Llama-3.3-70B-Instruct"
65
  tokenizer = AutoTokenizer.from_pretrained(model_id)
66
  model = AutoModelForCausalLM.from_pretrained(
67
  model_id,