Spaces:
Runtime error
Runtime error
Commit
·
30626c1
1
Parent(s):
59324e3
Remove flag
Browse files
app.py
CHANGED
|
@@ -27,7 +27,6 @@ tokenizer = AutoTokenizer.from_pretrained("bigcode/starcoderbase-3b")
|
|
| 27 |
vardecoder_model = AutoModelForCausalLM.from_pretrained(
|
| 28 |
"ejschwartz/resym-vardecoder",
|
| 29 |
torch_dtype=torch.bfloat16,
|
| 30 |
-
low_cpu_mem_usage=True,
|
| 31 |
device_map="cuda",
|
| 32 |
)
|
| 33 |
print("Loaded vardecoder model successfully.")
|
|
@@ -42,7 +41,6 @@ try:
|
|
| 42 |
fielddecoder_model = AutoModelForCausalLM.from_pretrained(
|
| 43 |
"ejschwartz/resym-fielddecoder",
|
| 44 |
torch_dtype=torch.bfloat16,
|
| 45 |
-
low_cpu_mem_usage=True,
|
| 46 |
device_map="cuda",
|
| 47 |
)
|
| 48 |
logger.info("Successfully loaded fielddecoder model")
|
|
|
|
| 27 |
vardecoder_model = AutoModelForCausalLM.from_pretrained(
|
| 28 |
"ejschwartz/resym-vardecoder",
|
| 29 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 30 |
device_map="cuda",
|
| 31 |
)
|
| 32 |
print("Loaded vardecoder model successfully.")
|
|
|
|
| 41 |
fielddecoder_model = AutoModelForCausalLM.from_pretrained(
|
| 42 |
"ejschwartz/resym-fielddecoder",
|
| 43 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 44 |
device_map="cuda",
|
| 45 |
)
|
| 46 |
logger.info("Successfully loaded fielddecoder model")
|