BIGAI-models
/

MagicText-1.2_fine-tuning

Text Generation

character-level

Model card Files Files and versions

BIGAI-models commited on 18 days ago

Commit

f030a27

·

verified ·

1 Parent(s): 31187fe

Update example.py

Files changed (1) hide show

example.py +1 -1

example.py CHANGED Viewed

@@ -20,7 +20,7 @@ class AI(nn.Module):
         x = self.linai(x)
         return x
 model = AI()
-model.load_state_dict(torch.load('C:\\Users\\name\\Downloads\\pytorch_mode1.pth'))
 model.eval()
 for _ in range(10):

         x = self.linai(x)
         return x
 model = AI()
+model.load_state_dict(torch.load('C:\\Users\\name\\Downloads\\pytorch_model.pth'))
 model.eval()
 for _ in range(10):