bobber commited on
Commit
4482b12
·
verified ·
1 Parent(s): 1e7b613

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,12 +5,12 @@ import torch
5
 
6
  #Qwen/Qwen2.5-14B-Instruct-1M
7
  #Qwen/Qwen2-0.5B
8
- model_name = "bobber/Qwen-0.5B-GRPO"
9
  subfolder = "Qwen-0.5B-GRPO/checkpoint-1868"
10
 
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_name,
13
- subfolder=subfolder,
14
  torch_dtype=torch.bfloat16,
15
  device_map="auto"
16
  )
 
5
 
6
  #Qwen/Qwen2.5-14B-Instruct-1M
7
  #Qwen/Qwen2-0.5B
8
+ model_name = "simplescaling/s1.1-32B"
9
  subfolder = "Qwen-0.5B-GRPO/checkpoint-1868"
10
 
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_name,
13
+ # subfolder=subfolder,
14
  torch_dtype=torch.bfloat16,
15
  device_map="auto"
16
  )