Spaces:

cdcvd
/

Evalution_resume

Running

cdcvd commited on Sep 2

Commit

1440e91

•

1 Parent(s): 158022c

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -93,7 +93,7 @@ import torch
 def evaluate_with_qwen(resume_text, job_description):
     # بارگذاری مدل و توکنایزر
     model = AutoModelForCausalLM.from_pretrained(
-        "Qwen/Qwen2-72B-Instruct",
         torch_dtype="auto",
         device_map="auto"
     ).to("cuda" if torch.cuda.is_available() else "cpu")

 def evaluate_with_qwen(resume_text, job_description):
     # بارگذاری مدل و توکنایزر
     model = AutoModelForCausalLM.from_pretrained(
+        "Qwen/Qwen2-1.5B-Instruct",
         torch_dtype="auto",
         device_map="auto"
     ).to("cuda" if torch.cuda.is_available() else "cpu")