Spaces:
Sleeping
Sleeping
Commit
·
ad43a4e
1
Parent(s):
7a6311a
test deepseek
Browse files
app.py
CHANGED
@@ -53,9 +53,7 @@ def deepseek(
|
|
53 |
system_message,
|
54 |
max_tokens,
|
55 |
temperature,
|
56 |
-
top_p
|
57 |
-
image_uploaded,
|
58 |
-
file_uploaded):
|
59 |
model_name = "deepseek-ai/deepseek-math-7b-instruct"
|
60 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
61 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|
|
|
53 |
system_message,
|
54 |
max_tokens,
|
55 |
temperature,
|
56 |
+
top_p):
|
|
|
|
|
57 |
model_name = "deepseek-ai/deepseek-math-7b-instruct"
|
58 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
59 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|