alexputhiyadom commited on
Commit
f44cbc9
·
verified ·
1 Parent(s): 765a0eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ class SmartLLMAgent:
15
  print("🔁 Loading SmartLLMAgent with Mistral-7B...")
16
  self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
17
  self.model = AutoModelForCausalLM.from_pretrained(
18
- "mistralai/Mistral-7B-Instruct-v0.2",
19
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
20
  device_map="auto"
21
  )
 
15
  print("🔁 Loading SmartLLMAgent with Mistral-7B...")
16
  self.tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
17
  self.model = AutoModelForCausalLM.from_pretrained(
18
+ "HuggingFaceH4/zephyr-7b-beta",
19
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
20
  device_map="auto"
21
  )