Hugging Face's logo

StevesInfinityDrive
/
JoyCloud

Safetensors
qwen2
Model card Files
xet
 Community
StevesInfinityDrive commited on
Commit
9ab0f08
·
verified ·
1 Parent(s): dddd023

Update src/model_loader.py

Browse files
Files changed (1) hide show
  1. src/model_loader.py +1 -1
src/model_loader.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  # Define the model path (adjust as needed)
5
- MODEL_PATH = "StevesInfinityDrive/JoyCloud" # Your fine-tuned model path
6
 
7
  # System prompt for guiding model behavior
8
  DEFAULT_PROMPT = """<|system|>
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  # Define the model path (adjust as needed)
5
+ MODEL_PATH = "model_files" # Your fine-tuned model path
6
 
7
  # System prompt for guiding model behavior
8
  DEFAULT_PROMPT = """<|system|>