nvidia
/

Nemotron-4-Mini-Hindi-4B-Base

Model card Files Files and versions Community

ravirajoshi commited on Oct 22, 2024

Commit

1710dbe

·

verified ·

1 Parent(s): aae0ef3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -70,7 +70,7 @@ dtype  = torch.bfloat16
 model  = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=dtype, device_map=device)
 # Prepare the input text
-prompt = 'भारत की संस्कृति के बारे में बताएं।'
 inputs = tokenizer.encode(prompt, return_tensors='pt').to(model.device)
 # Generate the output

 model  = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=dtype, device_map=device)
 # Prepare the input text
+prompt = 'भारत की राजधानी क्या है?'
 inputs = tokenizer.encode(prompt, return_tensors='pt').to(model.device)
 # Generate the output