Merge branch 'main' of
Browse files
app.py
CHANGED
|
@@ -32,4 +32,4 @@ peft_model_base = AutoModelForCausalLM.from_pretrained(model_name, quantization_
|
|
| 32 |
peft_model = PeftModel.from_pretrained(peft_model_base,
|
| 33 |
'red1xe/Llama-2-7B-codeGPT',
|
| 34 |
torch_dtype=torch.bfloat16,
|
| 35 |
-
is_trainable=False)
|
|
|
|
| 32 |
peft_model = PeftModel.from_pretrained(peft_model_base,
|
| 33 |
'red1xe/Llama-2-7B-codeGPT',
|
| 34 |
torch_dtype=torch.bfloat16,
|
| 35 |
+
is_trainable=False)
|