Hugging Face's logo

Spaces:
gaur3009
/
training_and_generation
Sleeping

App Files Community
gaur3009 commited on
Commit
1e7038e
·
verified ·
1 Parent(s): 637d90b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ index.add(embeddings_np)
35
  # -------------------------------
36
  # Load LLM (Phi-4-mini)
37
  # -------------------------------
38
- model_name = "microsoft/phi-4-mini"
39
  tokenizer = AutoTokenizer.from_pretrained(model_name)
40
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float32, device_map="auto")
41
 
 
35
  # -------------------------------
36
  # Load LLM (Phi-4-mini)
37
  # -------------------------------
38
+ model_name = "microsoft/phi-2"
39
  tokenizer = AutoTokenizer.from_pretrained(model_name)
40
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float32, device_map="auto")
41