Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,9 @@ login(token=os.environ["HF_ACCESS_TOKEN"]) #for read access!!!!
|
|
85 |
#base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
|
86 |
#base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
|
87 |
#base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
|
88 |
-
base_model="gpt2-xl"
|
|
|
|
|
89 |
|
90 |
#Tokenizer und Model laden
|
91 |
tokenizer,model,device = load_tokenizer_and_model(base_model, False)
|
|
|
85 |
#base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
|
86 |
#base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
|
87 |
#base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
|
88 |
+
#base_model="gpt2-xl"
|
89 |
+
base_model = "stabilityai/stablecode-instruct-alpha-3b" #load_8bit = ?
|
90 |
+
# Load model directly
|
91 |
|
92 |
#Tokenizer und Model laden
|
93 |
tokenizer,model,device = load_tokenizer_and_model(base_model, False)
|