Hugging Face's logo

Spaces:
dicksinyass
/
frtyh
Sleeping

App Files Community
dicksinyass commited on
Commit
998a82b
·
verified ·
1 Parent(s): 318bf1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ def load_model_and_tokenizer(hf_token=None):
8
  if hf_token:
9
  login(token=hf_token) # Use the provided token
10
 
11
- tokenizer = AutoTokenizer.from_pretrained("ryanmarten/Sky-T1-32B-Preview-5k-1-epoch", trust_remote_code=True)
12
- model = AutoModelForCausalLM.from_pretrained("ryanmarten/Sky-T1-32B-Preview-5k-1-epoch", trust_remote_code=True)
13
  return tokenizer, model, "Model and tokenizer loaded successfully!"
14
  except Exception as e:
15
  return None, None, f"Error loading model/tokenizer: {e}"
 
8
  if hf_token:
9
  login(token=hf_token) # Use the provided token
10
 
11
+ tokenizer = AutoTokenizer.from_pretrained(ryanmarten/Sky-T1-32B-Preview-5k-1-epoch, trust_remote_code=True)
12
+ model = AutoModelForCausalLM.from_pretrained(ryanmarten/Sky-T1-32B-Preview-5k-1-epoch, trust_remote_code=True)
13
  return tokenizer, model, "Model and tokenizer loaded successfully!"
14
  except Exception as e:
15
  return None, None, f"Error loading model/tokenizer: {e}"